2002: [Hnoi2010]Bounce 弹飞绵羊
Time Limit: 10 Sec Memory Limit: 259 MBSubmit: 10814 Solved: 5562
[Submit][Status][Discuss]
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
4
1 2 1 1
3
1 1
2 1 1
1 1
1 2 1 1
3
1 1
2 1 1
1 1
Sample Output
2
3
3
一看到此题就想到了递推+分块，写完后莫名WA了2次后来才发现修改时能将绵羊弹飞的节点的to数组没归0.
后来看题解发现正解是动态树（LCT）蒟蒻的我并不会写，这里贴出分块代码
1 #include<iostream> 2 #include<cstring> 3 #include<cstdio> 4 #include<cstdlib> 5 #include<cmath> 6 #include<algorithm> 7 using namespace std; 8 int read() 9 { 10 char ch=getchar();int x=0,f=1; 11 while(!isdigit(ch)){if(ch=='-')f=-1; ch=getchar();} 12 while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();} 13 return x*f; 14 } 15 int n; 16 int a[200002]; 17 int belong[200002],to[200002],b[200002]; 18 int size; 19 int main() 20 { 21 n=read(); 22 size=sqrt(n); 23 for(int i=1;i<=n;i++) 24 { 25 a[i]=read(); 26 belong[i]=(i-1)/size+1; 27 } 28 for(int i=n;i>=1;i--) 29 { 30 if(i+a[i]>n) b[i]=1; 31 else if(belong[i]==belong[i+a[i]]){b[i]=b[i+a[i]]+1;to[i]=to[i+a[i]];} 32 else{b[i]=1;to[i]=i+a[i];} 33 } 34 int m=read(); 35 while(m--) 36 { 37 int ask=read(); 38 if(ask==1) 39 { 40 int now=read(); 41 now++; 42 int ans=0; 43 while(now) 44 { 45 ans+=b[now]; 46 now=to[now]; 47 } 48 printf("%d ",ans); 49 } 50 else 51 { 52 int now=read(),c=read(); 53 now++; 54 a[now]=c; 55 int bn=belong[now]; 56 for(int i=now;belong[i]==bn;i--) 57 { 58 if(i+a[i]>n) b[i]=1,to[i]=0; 59 else if(belong[i]==belong[i+a[i]]){b[i]=b[i+a[i]]+1;to[i]=to[i+a[i]];} 60 else{b[i]=1;to[i]=i+a[i];} 61 } 62 } 63 } 64 }