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。
 
以上。