BSOJ3175:弹飞绵羊 LCT

3175 -- 【HNOI2010】弹飞绵羊

Description

　　Lostmonkey发明了一种超级反弹装置。

　　为了在绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿一条直线摆放 n个反弹装置，并按从前往后的方式将反弹装置依次编号为 0 到 n-1，对 0≤i≤n-1，为第 i 个反弹装置设定了初始弹力系数 ki，当绵羊落到第 i 个反弹装置上时，它将被往后弹出 ki 步，即落到第 i+ki 个反弹装置上，若不存在第i+ki个反弹装置，则绵羊被弹飞。

　　绵羊想知道： 从第i个反弹装置开始， 它被弹出几次 （含被弹飞的那次）后会被弹飞。为使游戏更有趣，Lostmonkey 还可以修改某个反弹装置的弹力系数，但任何时候弹力系数均为正整数。

Input

　　输入文件第一行是一个整数n，表示地上摆放n个反弹装置，

　　输入文件第二行是用空格隔开的n个正整数k0,k1,…,kn-1，分别表示n个反弹装置的初始弹力系数。

　　输入文件第三行是一个正整数m，表示后面还有m行输入数据。

　　接下来的m行，每行至少有用空格隔开的两个整数i和j，若i=1，则你要输出从第j个反弹装置开始，被弹出几次后会被弹飞；若i=2，则该行有用空格隔开的三个整数i，j和k，表示第j个反弹装置的弹力系数被修改为k。

　　输入的数据保证20%的数据满足n,m≤10000。100%的数据满足n≤200000,m≤100000

Output

　　输出文件output.txt包含的行数等于输入文件最后m行中i=1的行数。第h行输出一个整数，表示对输入中给出的第h个求弹出次数的问题，基于n个反弹装置当时的弹力系数，求出的弹出次数。

Sample Input

4       

1 2 1 1 

3 

1 1 

2 1 1 

1 1 

Sample Output

2

3

建树：题中输入每个弹射装置的距离，把单个弹射装置视作单个节点，如果当前位置+弹射距离>已有装置，则将当前位置连向空节点n+1，否则连向上述之和的节点，注意加偏移。

破题：初读此题， 修改弹射距离，即修改儿子节点，询问弹射次数，可以记录size，每次询问的是n+1到当前节点的size-1.由此可见，可以用动态树解。

详细： 1.修改节点——删除当前儿子节点，连接新点，注意判断超界。
              2.询问size——由于询问的是 当前节点->n+1的儿子数，即节点到n+1这一条链的size-1，需要把n+1变为根，构造当前节点到根的链为一棵Splytree，把当前节点旋为Slaytree树根，输出size-1（不！能！在！Pushup里把tree[x].size设为0，然后直接输出size，愚蠢的行为。）

Talk is cheap,show me the code.
 

#include<iostream>#include<cstdio>#include<cstring>using namespace std;int n,m;struct Splaytree{int l,r,fa,size;bool rev,onp;}tree[200005*3];int rec[200005];void Pushdown(int x){if(tree[x].rev){swap(tree[x].l,tree[x].r);tree[tree[x].l].rev^=1;tree[tree[x].r].rev^=1;tree[x].rev=0;}}void Pushup(int x){    tree[x].size=1;if(tree[x].l)tree[x].size+=tree[tree[x].l].size;    if(tree[x].r)tree[x].size+=tree[tree[x].r].size;}void leftrotate(int x){int y=tree[x].fa;tree[y].r=tree[x].l;if(tree[x].l)tree[tree[x].l].fa=y;tree[x].l=y;tree[x].fa=tree[y].fa;if(tree[y].onp){if(y==tree[tree[y].fa].l)tree[tree[y].fa].l=x;else tree[tree[y].fa].r=x;}tree[y].fa=x;if(tree[y].onp==0){tree[y].onp=1;tree[x].onp=0;}Pushup(y);}void rightrotate(int x){int y=tree[x].fa;tree[y].l=tree[x].r;if(tree[x].r)tree[tree[x].r].fa=y;tree[x].r=y;tree[x].fa=tree[y].fa;if(tree[y].onp){if(y==tree[tree[y].fa].l)tree[tree[y].fa].l=x;else tree[tree[y].fa].r=x;}tree[y].fa=x;if(tree[y].onp==0){tree[y].onp=1;tree[x].onp=0;}Pushup(y);}void Splay(int x){Pushdown(x);while(tree[x].onp){int y=tree[x].fa,z=tree[y].fa;if(tree[y].onp)Pushdown(z);Pushdown(y);Pushdown(x);if(tree[y].onp==0){if(tree[y].l==x)rightrotate(x);else leftrotate(x);break;}if(x==tree[y].l){if(y==tree[z].l)rightrotate(y),rightrotate(x);else rightrotate(x),leftrotate(x);}else {if(y==tree[z].l)leftrotate(x),rightrotate(x);else leftrotate(y),leftrotate(x);}}Pushup(x);}void Access(int x){int v=0;while(x){Splay(x);tree[tree[x].r].onp=0;tree[v].onp=1;tree[x].r=v;v=x;x=tree[x].fa;}}void Makeroot(int x){Access(x);Splay(x);        tree[x].rev^=1;}void Link(int v,int w){Makeroot(v);tree[v].fa=w;}void Cut(int v,int w){Makeroot(w);Access(v);Splay(v);tree[tree[v].l].fa=0;tree[tree[v].l].onp=0;tree[v].l=0;}void init(){scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%d",&rec[i]);Link(i,min(n+1,i+rec[i]));}scanf("%d",&m);}void solve(){int cmd;while(m--){int j,k;scanf("%d",&cmd);if(cmd==1){scanf("%d",&j);j++;Makeroot(n+1);Access(j);Splay(j);printf("%d\n",tree[j].size-1);}else{scanf("%d%d",&j,&k);j++;   Cut(j,min(j+rec[j],n+1));rec[j]=k;Link(j,min(j+rec[j],n+1));}}}int main(){init();solve();return 0;}

写的时候出现的错误：
1.Pushup()：忘记把当前节点的size设为1，答案全部输出-1.
2.Splay()：忘记在开始时Pushdown(x)，导致全体超时.
3.建树&修改时，判断是否超界写了一大堆if else,最后发现只需要一个min(j+rec[j],n+1)就好了。
4.询问操作时没有Makeroot(n+1);Access(j);Splay(j)，直接输出tree[j].size-1，导致样例全部输出0并且全部WA。
 
以上。