题意
城市\(i\)的海拔高度为\(H_i\)(各不相同)。定义距离为海拔差的绝对值
小\(A\)和小\(B\)轮流开车。从\(S\)起,一直向东行驶。
小\(A\)会选择第二近的城市作为目的地。小\(B\)选择一个最近的城市作为目的地。(如果当前城市到两个城市的距离相同,则认为离海拔低的那个城市更近)。如果无法再开了,或者到达目的地会使行驶的总距离超出\(X\)公里,他们就会结束旅行。
单次询问:给定\(X\),问从哪个城市出发,小\(A\)行的路程与小\(B\)行的路程的比值最小
多次询问:给定\(S\)和\(X\),问小\(A\)行的路程与小\(B\)行的路程
\(1 \leq N \leq 10^5,1 \leq M \leq 10^5\)
思路
首先肯定要解决一个问题,小\(A\)和小\(B\)的目的地在哪里。
先排序,然后从第一个城市开始,比较\(i-1,i-2,i+1,i+2\)处距离。
很显然,由于是第一个点,这个时候找到的任何点一定在它东边。
同样的,找完第一个点后将第一个点删除,那么第二个点自然成为了第一个点,依次下去,便在\(O(n)\)的复杂度内完成了预处理。
接下来,就可以想到用倍增来解决。
用\(f[i][j]\)表示从\(j\)出发第\(2^i\)天到达的城市,发现除了\(i=0\)时是\(A\)在开其他都是\(B\)开,所以转移就很简单了\(f[i][j]=f[i-1][f[i-1][j]];\)
那么求出了路径,距离只要减一减就好了。
用\(g[i][j][0]\)表示从\(j\)出发第\(2^i\)天\(A\)开的,用\(g[i][j][1]\)表示从\(j\)出发第\(2^i\)天\(B\)开的,转移:
\(g[i][j][0]=g[i-1][j][0]+g[i-1][f[i-1][j]][0];\)
\(g[i][j][1]=g[i-1][j][1]+g[i-1][f[i-1][j]][1];\)
接下来的询问只要每次倍增凑距离就好了
对于第一个询问枚举出发点再比较
时间复杂度\(O((n+m)logn)\)
#include <bits/stdc++.h>
const int N=100005;
double INF=1000000000;
using std::sort;
int a[N],b[N],Next[N],last[N],c[N],n,m,f[21][N],x,ans=0,s;
long long h[N],suma,sumb,g[21][N][2];
bool cmp(int x,int y){
return h[x]<h[y];
}
int d(int x,int y){
return abs(h[x]-h[y]);
}
void solve(int x,int s){
for (int i=20;i>=0;i--)
if (g[i][s][0]+g[i][s][1]<=1ll*x){
x-=g[i][s][0]+g[i][s][1];
suma+=g[i][s][0];
sumb+=g[i][s][1];
if (s==f[i][s]) return;
s=f[i][s];
}
}
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++) scanf("%lld",&h[i]),c[i]=i;
h[0]=-200000000000;
sort(c+1,c+n+1,cmp);
for (int i=1;i<=n;i++) last[c[i]]=c[i-1],Next[c[i]]=c[i+1];
for (int i=1;i<=n;i++){
int x1=last[i],x2=last[x1],x3=Next[i],x4=Next[x3];
int dis1=abs(h[i]-h[x1]),dis2=abs(h[i]-h[x2]),dis3=abs(h[i]-h[x3]),dis4=abs(h[i]-h[x4]);
if (dis1<dis3 ||(dis1==dis3 && h[x1]<h[x3])) b[i]=x1; else b[i]=x3;
if (dis2<dis3 ||(dis2==dis3 && h[x2]<h[x3])) a[i]=x2;
else if (dis4<dis1 ||(dis4==dis1 && h[x4]<h[x1])) a[i]=x4;
else if (dis1<dis3 ||(dis1==dis3 && h[x1]<h[x3])) a[i]=x3;
else a[i]=x1;
if (a[i]==0) a[i]=i;
if (b[i]==0) b[i]=i;
last[Next[i]]=last[i],Next[last[i]]=Next[i];
}
for (int i=1;i<=n;i++){
f[0][i]=a[i];if (a[i]!=i) f[1][i]=b[a[i]];else f[1][i]=i;
g[0][i][0]=d(i,f[0][i]),g[1][i][0]=g[0][i][0];
if (a[i]!=i)g[1][i][1]=d(a[i],b[a[i]]);
}
for (int i=2;i<=20;i++)
for (int j=1;j<=n;j++){
f[i][j]=f[i-1][f[i-1][j]];
g[i][j][0]=g[i-1][j][0]+g[i-1][f[i-1][j]][0];
g[i][j][1]=g[i-1][j][1]+g[i-1][f[i-1][j]][1];
}
scanf("%d",&x);
double now=INF;
for (int i=1;i<=n-2;i++){
suma=0,sumb=0;
solve(x,i);
if (now==INF && sumb==0 && h[i]>h[ans]) ans=i;
if (sumb==0) continue;
if (suma*1.0/sumb<now ||(suma*1.0/sumb==now && h[i]>h[ans])){
now=suma*1.0/sumb;
ans=i;
}
}
printf("%d\n",ans);
scanf("%d",&m);
for (int i=1;i<=m;i++){
scanf("%d%d",&s,&x);
suma=0,sumb=0;
solve(x,s);
printf("%lld %lld\n",suma,sumb);
}
}
后记
要注意\(h[0]\)的初始化,原来没仔细看数据结果爆了\(4\)发