思路:
先预处理出每个城市出发,小A小B开车到达的下一个城市。
然后倍增优化DP:
{
f
0
,
i
,
,
j
=
f
0
,
f
0
,
i
,
j
−
1
,
j
−
1
d
i
s
0
,
i
,
j
=
d
i
s
0
,
i
,
j
−
1
+
d
i
s
0
,
f
0
,
i
,
j
−
1
,
j
−
1
d
i
s
1
,
i
,
j
=
d
i
s
1
,
i
,
j
−
1
+
d
i
s
1
,
f
0
,
i
,
j
−
1
,
j
−
1
\begin{cases} f_{0,i,,j}=f_{0,f_{0,i,j-1},j-1} \\dis_{0,i,j}=dis_{0,i,j-1}+dis_{0,f_{0,i,j-1},j-1} \\ dis_{1,i,j}=dis_{1,i,j-1}+dis_{1,f_{0,i,j-1},j-1} \end{cases}
⎩⎪⎨⎪⎧f0,i,,j=f0,f0,i,j−1,j−1dis0,i,j=dis0,i,j−1+dis0,f0,i,j−1,j−1dis1,i,j=dis1,i,j−1+dis1,f0,i,j−1,j−1
特别的,当
j
=
1
j=1
j=1 时,有:
{
f
0
,
i
,
,
1
=
f
1
,
f
0
,
i
,
0
,
0
d
i
s
0
,
i
,
1
=
d
i
s
0
,
i
,
0
d
i
s
1
,
i
,
1
=
d
i
s
1
,
f
0
,
i
,
0
,
0
\begin{cases} f_{0,i,,1}=f_{1,f_{0,i,0},0} \\dis_{0,i,1}=dis_{0,i,0} \\ dis_{1,i,1}=dis_{1,f_{0,i,0},0} \end{cases}
⎩⎪⎨⎪⎧f0,i,,1=f1,f0,i,0,0dis0,i,1=dis0,i,0dis1,i,1=dis1,f0,i,0,0
(因为当跳两步时,两段分别是由不同人跳的,所以去反)
c o d e code code
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
long long n, m;
struct node
{
long long hi, id;
long long pre, nxt;
}h[100010];
long long pos[100010], ga[100010], gb[100010], f[100010][20][2], disa[100010][20][2], disb[100010][20][2];
bool cmp(node x, node y)
{
return x.hi<y.hi;
}
long long Zi_O(long long i, long long a, long long b)
{
if(!a)
return h[b].id;
if(!b)
return h[a].id;
if(h[i].hi-h[a].hi<=h[b].hi-h[i].hi)
return h[a].id;
else
return h[b].id;
}
void decade(long long i)
{
if(h[i].nxt)
h[h[i].nxt].pre=h[i].pre;
if(h[i].pre)
h[h[i].pre].nxt=h[i].nxt;
}
long long la, lb;
void Ex_aid(long long x, long long maxx)
{
la=0, lb=0;
long long k=0;
for(long long i=log2(n); i>=0; i--)
{
if(f[x][i][k]&&disa[x][i][k]+disb[x][i][k]<=maxx)
{
maxx-=disa[x][i][k]+disb[x][i][k];
la+=disa[x][i][k], lb+=disb[x][i][k];
if(i==0)
k=1-k;
x=f[x][i][k];
}
}
}
int main()
{
scanf("%lld", &n);
for(long long i=1; i<=n; i++)
{
scanf("%lld", &h[i].hi);
h[i].id=i;
}
sort(h+1, h+1+n, cmp);
for(long long i=1; i<=n; i++)
{
pos[h[i].id]=i;
h[i].pre=i-1;
h[i].nxt=i+1;
}
h[1].pre=h[n].nxt=0;
for(long long i=1; i<n; i++)
{
long long p=pos[i], p1=h[p].pre, p2=h[p].nxt;
if(p1&&(h[p].hi-h[p1].hi<=h[p2].hi-h[p].hi||!p2))
gb[i]=h[p1].id, ga[i]=Zi_O(p, h[p1].pre, p2);
else
gb[i]=h[p2].id, ga[i]=Zi_O(p, p1, h[p2].nxt);
decade(p);
}
for(long long i=1; i<=n; i++)
{
if(ga[i])
{
f[i][0][0]=ga[i];
disa[i][0][0]=abs(h[pos[i]].hi-h[pos[ga[i]]].hi);
disb[i][0][0]=0;
}
if(gb[i])
{
f[i][0][1]=gb[i];
disa[i][0][1]=0;
disb[i][0][1]=abs(h[pos[i]].hi-h[pos[gb[i]]].hi);
}
}
for(long long j=1; j<=log2(n); j++)
{
for(long long i=1; i<=n; i++)
{
for(long long k=0; k<=1; k++)
{
int l=k;
if(j==1)
l=1-k;
if(f[i][j-1][k])
f[i][j][k]=f[f[i][j-1][k]][j-1][l];
if(f[i][j][k])
disa[i][j][k]=disa[i][j-1][k]+disa[f[i][j-1][k]][j-1][l],
disb[i][j][k]=disb[i][j-1][k]+disb[f[i][j-1][k]][j-1][l];
}
}
}
long long x, ansa=1, ansb=0, p=0;
scanf("%lld", &x);
for(long long i=1; i<=n; i++)
{
Ex_aid(i, x);
if(!lb)
la=1;
if(la*ansb<lb*ansa||(la*ansb==lb*ansa&&h[pos[i]].hi>h[pos[p]].hi))
ansa=la, ansb=lb, p=i;
}
printf("%lld\n", p);
scanf("%lld", &m);
while(m--)
{
long long s, x;
scanf("%lld%lld", &s, &x);
Ex_aid(s, x);
printf("%lld %lld\n", la, lb);
}
return 0;
}