BZOJ2093: [Poi2010]Frog

来源:互联网 发布:自由行 知乎 编辑:程序博客网 时间:2024/05/22 11:53

题目大意:给一条线上n个石头,青蛙从跳m步,每次跳到离当前第k近的石头,求以所有点为起点最终会跳到哪


很关键的一步是求出从每个点跳一步会去到哪里

这个可以用两个指针维护一段区间,代表前k近的石头的区间,然后从左到右扫一遍,指针也跟着平移,就能O(N)扫出来了

之后的部分可以在环套树森林上DFS来O(N)出解,不过这个有点麻烦,于是我就写了个倍增(环套树倍增也算树倍增吧?)

不过倍增还卡内存,所以要用一个类似滚动数组的东西.....


#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#define N 1000010using 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]);}

0 0
原创粉丝点击