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,否则连向上述之和的节点,注意加偏移。
建树:题中输入每个弹射装置的距离,把单个弹射装置视作单个节点,如果当前位置+弹射距离>已有装置,则将当前位置连向空节点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。
以上。
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。
以上。