P1081 [NOIP2012 提高组] 开车旅行

思路：

首先令 \(nxt1_i\) 表示右侧最近的城市距离（\(id1_i\) 为编号），令 \(nxt2_i\) 表示右侧第二近的城市编号（\(id2_i\) 为编号）；可以使用 set 找出离这个城市最近的 \(4\) 个城市（前面两个，后面两个）。

定义：

• \(f_{i,j}\) 表示从 \(i\) 点出发走 \(2^j\) 轮最后到达的位置。

• \(dp1_{i,j}\) 表示从 \(i\) 点出发走 \(2^j\) 轮最后 A 走过的距离。

• \(dp2_{i,j}\) 表示从 \(i\) 点出发走 \(2^j\) 轮最后 B 走过的距离。

初始化：

\[f_{i,0} = id1_{id2_i} \]

\[dp1_{i,0} = nxt2_{i} \]

\[dp2_{i,0} = nxt1_{id2_i} \]

状态转移方程为：

\[f_{i,j} = f_{f_{i,j-1},j-1} \]

\[dp1_{i,j} = dp1_{i,j-1} + dp1_{f_{i,j-1},j-1} \]

\[dp2_{i,j} = dp2_{i,j-1} + dp2_{f_{i,j-1},j-1} \]

此时对于询问 \(1\) 和询问 \(2\)：

• 本质上是求出从每个城市出发后 \(A\) 走的距离与 \(B\) 走的距离。

• 那么考虑从高位到低位贪心，即设当前跳到了 \(s\) 点，若 \(dp1_{s,i} + dp2_{s,i} \le x\)，可以从 \(s\) 跳到 \(f_{s,i}\)，需要令 \(x \gets x – (dp1_{s,i} + dp2_{s,i})\)，然后继续遍历 \(i-1\) 位。

• 因为是 A 先开车，所以 A 可能会在最后一轮结束后还能再开上一次，需要特判。

时间复杂度为 \(O((N+Q) \log N)\)。

完整代码：

#include<bits/stdc++.h>
#define Add(x,y) (x+y>=mod)?(x+y-mod):(x+y)
#define lowbit(x) x&(-x)
#define pi pair<ll,ll>
#define pii pair<ll,pair<ll,ll>>
#define iip pair<pair<ll,ll>,ll>
#define ppii pair<pair<ll,ll>,pair<ll,ll>>
#define fi first
#define se second
#define full(l,r,x) for(auto it=l;it!=r;it++) (*it)=x
#define Full(a) memset(a,0,sizeof(a))
#define open(s1,s2) freopen(s1,"r",stdin),freopen(s2,"w",stdout);
using namespace std;
typedef double db;
typedef unsigned long long ull;
typedef long long ll;
bool Begin;
const ll N=1e5+10,M=18,INF=1e18;
inline ll read(){
    ll x=0,f=1;
    char c=getchar();
    while(c<'0'||c>'9'){
        if(c=='-')
          f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9'){
        x=(x<<1)+(x<<3)+(c^48);
        c=getchar();
    }
    return x*f;
}
inline void write(ll x){
	if(x<0){
		putchar('-');
		x=-x;
	}
	if(x>9)
	  write(x/10);
	putchar(x%10+'0');
}
struct Node{
	ll a,b;
	ll id;
	bool operator<(const Node&rhs)const{
		if(a!=rhs.a)
		  return a<rhs.a;
		return b<rhs.b;
	}
};
ll n,m,s,x,x0,s0,cnt,dis1,dis2,disa,disb;
ll a[N],nxt1[N],nxt2[N],id1[N],id2[N];
ll f[N][M],dp1[N][M],dp2[N][M];
Node h[N];
vector<Node> V;
multiset<Node> S;
void solve(ll s,ll x){
	dis1=dis2=0;
	for(int i=M-1;i>=0;i--){
		if(dp1[s][i]+dp2[s][i]<=x&&f[s][i]){
			dis1+=dp1[s][i],dis2+=dp2[s][i];
			x-=dp1[s][i]+dp2[s][i];
			s=f[s][i];
		}
	}
	if(nxt2[s]<=x)
	  dis1+=nxt2[s];
}
bool End;
int main(){
	n=read();
	for(int i=1;i<=n;i++){
		a[i]=read();
		h[i]={a[i],a[i],i};
	}
	S.insert({INF,INF,0});
	S.insert({INF-1,INF-1,0});
	S.insert({-INF,-INF,0});
	S.insert({-INF+1,-INF+1,0});
	for(int i=n;i>=1;i--){
		V.clear();
		V.push_back(*--S.lower_bound(h[i]));
		V.push_back(*--S.lower_bound(V[0]));
		V.push_back(*S.upper_bound(h[i]));
		V.push_back(*S.upper_bound(V[2]));
		for(auto &v:V)
		  v.a=abs(h[i].a-v.a);
		sort(V.begin(),V.end());
		nxt1[i]=V[0].a,nxt2[i]=V[1].a;
		id1[i]=V[0].id,id2[i]=V[1].id;
		cerr<<id1[i]<<' '<<id2[i]<<'\n';
		S.insert(h[i]);
	}
	for(int i=1;i<=n;i++){
		f[i][0]=id1[id2[i]];
		dp1[i][0]=nxt2[i];
		dp2[i][0]=nxt1[id2[i]];
	}
	for(int j=1;j<M;j++){
		for(int i=n;i>=1;i--){
			f[i][j]=f[f[i][j-1]][j-1];
			dp1[i][j]=dp1[i][j-1]+dp1[f[i][j-1]][j-1];
			dp2[i][j]=dp2[i][j-1]+dp2[f[i][j-1]][j-1];
		}
	}
	x0=read();
	for(int i=1;i<=n;i++){
		solve(i,x0);
		if(dis2&&(!s0||disa*dis2>disb*dis1)){
			s0=i;
			disa=dis1,disb=dis2;
		}
	}
	write(s0);
	putchar('\n');
	m=read();
	while(m--){
		s=read(),x=read();
		solve(s,x);
		write(dis1);
		putchar(' ');
		write(dis2);
		putchar('\n');
	}
	cerr<<'\n'<<abs(&Begin-&End)/1048576<<"MB";
	return 0;
}