题解 Luogu P1081 开车旅行

题意

给定 \(n\) 座城市,第 \(i\) 座城市有一个海拔高度 \(h_i\),且 \(h_i\) 两两不同。定义城市 \(i,j\) 的距离 \(dis(i,j)=|h_i-h_j|\),即城市 \(i,j\) 海拔之差的绝对值。

小 A 和小 B 将从开车城市 \(s\) 出发,并且最多行驶 \(x\) 公里就结束旅行。他们会轮流驾驶,并且小 A 驾驶第一天。如果某一天是小 A 驾驶,那么他会在编号大于当前城市的城市中,选择离 \(s\) 最第二近的一个城市作为目的地;否则,小 B 会在编号大于当前城市的城市中,选择离 \(s\) 最近的城市作为目的地。特别地,如果有多个这样的城市,他们总会选择编号最小的。

如果某一天任何人都无法按照自己的原则选择出当天的目的地,或者到达目的地会使总路程数超过 \(x\) 公里,那么旅行结束。请求出:

  • 对于一个给定的 \(x=x_0\),从哪一个城市出发,小 A 开车行驶的路程总数与小 B 行驶的路程总数的比值最小(如果小 B 的行驶路程为 \(0\),此时的比值可视为无穷大,且两个无穷大视为相等)。如果从多个城市出发,小 A 开车行驶的路程总数与小B 行驶的路程总数的比值都最小,则输出海拔最高的那个城市。

  • 对于给定的 \(m\) 个 \(s_i\) 和 \(x_i\),小 A 开车行驶的路程总数和小 B 开车行驶的路程总数。

\(1 \leq n,m \leq 10^5,-10^9 \leq h_i \leq 10^9,1 \leq s_i \leq n,0 \leq x_i \leq 10^9\)

题解

首先,观察到每个人在城市固定的情况下的选择是固定的。

因为只能够选择编号大于等于当前的城市,所以可选择的是一个后缀。倒序处理,用 set 维护出每个人会选择的城市(set::lower_bound 查前驱,后继,前驱的前驱,后继的后继)。

其次,我们观察到,在不考虑 \(x\) 的影响下,每一天的决策与前一天的决策无关。这启发我们可以将几段决策拼到一起成为最终的决策。考虑倍增优化。设 \(\operatorname{id}_{i,k,0/1}\) 表示从城市 \(i\) 出发,小 A / 小 B 先开车,经过 \(2^k\) 天之后,到达的城市编号。同时,我们设 \(\operatorname{lena}_{i,k,0/1},\operatorname{lenb}_{i,k,0/1}\),表示从城市 \(i\) 出发,小 A / 小 B 先开车,经过 \(2^k\) 天之后,小 A / 小 B 开车的路程总数。注意,上述的状态均不考虑 \(x\) 的影响

我们可以轻松地预处理出 \(\operatorname{id}_{i,0,0/1}\),对于 \(\operatorname{lena}\) 和 \(\operatorname{lenb}\) 同理。接下来,我们考虑转移。当 \(k>1\) 时,\(\operatorname{id}_{i,k,t}(t \in \{0,1\})\) 即 \(\operatorname{id}_{\operatorname{id}_{i,k-1,t},k-1,t}\)。而当 \(k=1\) 时,\(2^{k-1}=1\) 是一个奇数,所以前半部分是 \(t\) 先开车,而后半部分是 \(t \operatorname{xor} 1\) 先开车,需要注意。类似地,我们也可以得到 \(\operatorname{lena}\) 和 \(\operatorname{lenb}\) 的转移方程。有:

\[\operatorname{id}_{i,k,t} = \begin{cases}\operatorname{id}_{\operatorname{id}_{i,k-1,t},k-1,t} & k>1 \\ \operatorname{id}_{\operatorname{id}_{i,k-1,t},k-1,t \operatorname{xor} 1} & k = 1\end{cases} \]

\[\operatorname{lena}_{i,k,t}= \begin{cases}\operatorname{lena}_{i,k-1,t} + \operatorname{lena}_{\operatorname{id}_{i,k-1,t},k-1,t} & k>1 \\ \operatorname{lena}_{i,k-1,t} + \operatorname{lena}_{\operatorname{id}_{i,k-1,t},k-1,t \operatorname{xor} 1} & k=1\end{cases} \]

\(\operatorname{lenb}\) 同理。

对于第一问,枚举起点城市,然后倍增判断:如果从当前城市往前走 \(2^k\) 天路程总数不超过 \(x_0\),那么就往前走。并分别记录下两人开车的路程数,比较比值即可。

对于第二问同理,记录下两人开车的路程数即可。

\(n\) 和 \(m\) 同阶,则时间复杂度为 \(O(n \log n)\)。

# include <bits/stdc++.h>
# define int long long
const int N=100010,INF=0x7fffffff;

struct Node{
	int height,id;
	bool operator < (const Node &rhs) const{
		return height<rhs.height;
	}
};
std::multiset <Node> S;
int h[N];
int n;
int nex[N][2];
int nexid[N][21][2],lena[N][21][2],lenb[N][21][2]; // nexid[] 即文中的 id[]
inline int read(void){
	int res,f=1;
	char c;
	while((c=getchar())<'0'||c>'9')
		if(c=='-')f=-1;
	res=c-48;
	while((c=getchar())>='0'&&c<='9')
		res=res*10+c-48;
	return res*f;
}
inline int myabs(int x){
	return (x>0)?x:-x;
}
inline std::pair <int,int> calc(int st,int lim){
	int la=0,lb=0,now=st;
	for(int i=20;i>=0;--i){
		if(nexid[now][i][0]&&la+lb+lena[now][i][0]+lenb[now][i][0]<=lim){ // 如果 nexid[now][i][0] == 0 则不能从当前点往前走 2^i 天
			la+=lena[now][i][0],lb+=lenb[now][i][0];
			now=nexid[now][i][0];
		}
	}
	return std::make_pair(la,lb);
}
signed main(void){
	n=read();
	for(int i=1;i<=n;++i){
		h[i]=read();
	}
	S.insert((Node){-INF,0}),S.insert((Node){-INF,0});
	S.insert((Node){INF,0}),S.insert((Node){INF,0}); // 防止查不到而导致越界
	for(int i=n;i;--i){
		std::multiset <Node>::iterator l,r;
		Node L,R,LL,RR;
		l=r=S.lower_bound((Node){h[i],i});
		--l,L=*l,R=*r;
		--l,++r,LL=*l,RR=*r;
		if(myabs(h[i]-L.height)<=myabs(h[i]-R.height)){
			nex[i][1]=L.id;
			if(myabs(h[i]-LL.height)<=myabs(h[i]-R.height)){
				nex[i][0]=LL.id;
			}else{
				nex[i][0]=R.id;
			}
		}
		else if(R.height!=INF){
			nex[i][1]=R.id;
			if(myabs(h[i]-L.height)<=myabs(h[i]-RR.height)){
				nex[i][0]=L.id;
			}else{
				nex[i][0]=RR.id;
			}
		}
		S.insert((Node){h[i],i});
	}
	for(int i=1;i<=n;++i){
		nexid[i][0][0]=nex[i][0],nexid[i][0][1]=nex[i][1];
		lena[i][0][0]=myabs(h[i]-h[nex[i][0]]),lenb[i][0][1]=myabs(h[i]-h[nex[i][1]]);
	}
	for(int k=1;k<=20;++k){
		for(int i=1;i<=n;++i){
			for(int s=0;s<=1;++s){
				int mid=nexid[i][k-1][s];
				if(k==1){
					nexid[i][k][s]=nexid[mid][k-1][s^1];
					lena[i][k][s]=lena[i][k-1][s]+lena[mid][k-1][s^1];
					lenb[i][k][s]=lenb[i][k-1][s]+lenb[mid][k-1][s^1];
					continue;
				}
				nexid[i][k][s]=nexid[mid][k-1][s];
				lena[i][k][s]=lena[i][k-1][s]+lena[mid][k-1][s];
				lenb[i][k][s]=lenb[i][k-1][s]+lenb[mid][k-1][s];
			}
		}
	}
	int X=read();
	double minv=2e18;
	int id;
	for(int i=1;i<=n;++i){
		std::pair <int,int> ans=calc(i,X);
		int la=ans.first,lb=ans.second;
		double nowv=(lb)?((double)la/(double)lb):(1e18);
		if(minv>nowv||(minv==nowv&&h[id]<h[i])){
			id=i,minv=nowv;
		}
	}
	printf("%lld\n",id);
	int m=read();
	while(m--){
		int si=read(),xi=read();
		std::pair <int,int> ans=calc(si,xi);
		printf("%lld %lld\n",ans.first,ans.second);
	}
	return 0;
}
上一篇:Hive篇--相关概念和使用二


下一篇:OI卷题记录