Description
某天,Lostmonkey发明了一种超级弹力装置,为了在他的绵羊朋友面前显摆,他邀请小绵羊一起玩个游戏。游戏一开始,Lostmonkey在地上沿着一条直线摆上n个装置,每个装置设定初始弹力系数ki,当绵羊达到第i个装置时,它会往后弹ki步,达到第i+ki个装置,若不存在第i+ki个装置,则绵羊被弹飞。绵羊想知道当它从第i个装置起步时,被弹几次后会被弹飞。为了使得游戏更有趣,Lostmonkey可以修改某个弹力装置的弹力系数,任何时候弹力系数均为正整数。
Input
第一行包含一个整数n,表示地上有n个装置,装置的编号从0到n-1,接下来一行有n个正整数,依次为那n个装置的初始弹力系数。第三行有一个正整数m,接下来m行每行至少有两个数i、j,若i=1,你要输出从j出发被弹几次后被弹飞,若i=2则还会再输入一个正整数k,表示第j个弹力装置的系数被修改成k。对于20%的数据n,m<=10000,对于100%的数据n<=200000,m<=100000
Output
对于每个i=1的情况,你都要输出一个需要的步数,占一行。
Sample Input
1 2 1 1
3
1 1
2 1 1
1 1
Sample Output
23
正解:LCT或分块。
这题一眼看上去就是LCT,但是不会写。然后A过的人告诉我是分块,想了一会儿yy出来了。。
记录每个点跳到当前块的最后一个位置和跳到下一个块第一个位置的距离,这个从n到1递推就好。查询时每次跳一个块,所以最多跳sqrt(n)次。修改时只要修改当前点和在这个块中的前面的点,所以最多修改sqrt(n)次,那么总复杂度就是m*sqrt(n)。
//It is made by wfj_2048~
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#define inf (1<<30)
#define il inline
#define RG register
#define ll long long
#define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
using namespace std;
int k[400010],bl[400010],far[400010],dis[400010],n,m,block;
il int gi(){
RG int x=0,q=1; RG char ch=getchar(); while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
if (ch=='-') q=-1,ch=getchar(); while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar(); return q*x;
}
il void work(){
n=gi(),block=sqrt(n); for (RG int i=1;i<=n;++i) k[i]=gi(),bl[i]=(i-1)/block+1; m=gi();
for (RG int i=n+1;i<=2*n;++i) bl[i]=i;
for (RG int i=n;i;--i)
if (bl[i+k[i]]>bl[i]) far[i]=i,dis[i]=1;
else far[i]=far[i+k[i]],dis[i]=dis[i+k[i]]+1;
for (RG int i=1;i<=m;++i){
RG int type=gi(),x=gi()+1,ans=0;
if (type==1){ while (x<=n) ans+=dis[x],x=far[x]+k[far[x]]; printf("%d\n",ans); }
if (type==2){
RG int K=gi(),Bl=bl[x]; k[x]=K;
for (;bl[x]==Bl;--x)
if (bl[x+k[x]]>bl[x]) far[x]=x,dis[x]=1;
else far[x]=far[x+k[x]],dis[x]=dis[x+k[x]]+1;
}
}
return;
}
int main(){
File("bounce");
work();
return 0;
}
ps:2.19改,现在会写LCT了,弄一个LCT的代码吧。。
//It is made by wfj_2048~
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#define inf (1<<30)
#define il inline
#define RG register
#define ll long long
#define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
using namespace std;
int ch[200010][2],size[200010],lazy[200010],fa[200010],st[200010],k[200010],n,m;
il int gi(){
RG int x=0,q=1; RG char ch=getchar(); while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
if (ch=='-') q=-1,ch=getchar(); while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar(); return q*x;
}
il void pushdown(RG int x){ lazy[x]=0,lazy[ch[x][0]]^=1,lazy[ch[x][1]]^=1,swap(ch[x][0],ch[x][1]); return; }
il void pushup(RG int x){ if (!x) size[x]=0; else size[x]=size[ch[x][0]]+size[ch[x][1]]+1; return; }
il int isroot(RG int x){ return ch[fa[x]][0]!=x && ch[fa[x]][1]!=x; }
il void rotate(RG int x){
RG int y=fa[x],z=fa[y],k=(ch[y][0]==x); if (!isroot(y)) ch[z][ch[z][1]==y]=x; fa[x]=z;
ch[y][k^1]=ch[x][k],fa[ch[x][k]]=y,ch[x][k]=y,fa[y]=x,pushup(y),pushup(x); return;
}
il void splay(RG int x){
RG int top=0; st[++top]=x;
for (RG int i=x;!isroot(i);i=fa[i]) st[++top]=fa[i];
for (RG int i=top;i;--i) if (lazy[st[i]]) pushdown(st[i]);
while (!isroot(x)){
RG int y=fa[x],z=fa[y];
if (!isroot(y)){
if ((ch[z][0]==y)^(ch[y][0]==x)) rotate(x); else rotate(y);
}
rotate(x);
}
return;
}
il void access(RG int x){ RG int t=0; while (x){ splay(x),ch[x][1]=t,t=x,x=fa[x]; } return; }
il void makeroot(RG int x){ access(x),splay(x),lazy[x]^=1; return; }
il void link(RG int x,RG int y){ makeroot(x),fa[x]=y; return; }
il void cut(RG int x,RG int y){ makeroot(x),access(y),splay(y),ch[y][0]=fa[x]=0; return; }
il void work(){
n=gi(); RG int type,x,y; for (RG int i=1;i<=n+1;++i) size[i]=1;
for (RG int i=1;i<=n;++i) k[i]=gi(),link(i,min(i+k[i],n+1)); m=gi();
for (RG int i=1;i<=m;++i){
type=gi();
if (type==1){ x=gi()+1,makeroot(n+1),access(x),splay(x); printf("%d\n",size[ch[x][0]]); }
if (type==2) x=gi()+1,y=gi(),cut(x,min(x+k[x],n+1)),link(x,min(x+y,n+1)),k[x]=y;
}
return;
}
int main(){
File("bounce");
work();
return 0;
}