题目大意:给一条线上n个石头,青蛙从跳m步,每次跳到离当前第k近的石头,求以所有点为起点最终会跳到哪
很关键的一步是求出从每个点跳一步会去到哪里
这个可以用两个指针维护一段区间,代表前k近的石头的区间,然后从左到右扫一遍,指针也跟着平移,就能O(N)扫出来了
之后的部分可以在环套树森林上DFS来O(N)出解,不过这个有点麻烦,于是我就写了个倍增(环套树倍增也算树倍增吧?)
不过倍增还卡内存,所以要用一个类似滚动数组的东西.....
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000010
using namespace std;
long long a[N];
int b[N][2];
int ans[N];
int i,j,o,x,y;
main()
{
int n,k;
long long m;
scanf("%d%d%lld",&n,&k,&m);
for(i=1;i<=n;i++)
scanf("%lld",&a[i]);
b[1][0]=k+1;
int l=1,r=k+1;
for(i=2;i<=n;i++)
{
while(r<n&&a[i]-a[l]>a[r+1]-a[i]) l++,r++;
if(a[i]-a[l]<a[r]-a[i]) b[i][0]=r;
else b[i][0]=l;
}
for(i=1;i<=n;i++)
ans[i]=i;
for(i=0;(1LL<<i)<=m;i++)
{
o=i&1;
if(m&(1LL<<i))
for(j=1;j<=n;j++)
ans[j]=b[ans[j]][o];
for(j=1;j<=n;j++)
b[j][o^1]=b[b[j][o]][o];
}
printf("%d",ans[1]);
for(i=2;i<=n;i++)
printf(" %d",ans[i]);
}