本文版权归ljh2000和博客园共有,欢迎转载,但须保留此声明,并给出原文链接,谢谢合作。
本文作者:ljh2000
作者博客:http://www.cnblogs.com/ljh2000-jump/
转载请注明出处,侵权必究,保留最终解释权!
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
正解:LCT
解题报告:
LCT模板题,话说以前一直是用分块写来着...
我开始一直没弄清楚,为啥有根树可以支持换根操作,后来发现并没有关系,只要在询问的时候先把根换回来再询问就好了…
建立一个虚点,每次把查询点access之后,旋转到根,左子树size就是答案了。
//It is made by ljh2000
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <string>
#include <complex>
using namespace std;
typedef long long LL;
typedef long double LB;
typedef complex<double> C;
const double pi = acos(-1);
const int MAXN = 200011;
int n,m,tr[MAXN][2],tag[MAXN],val[MAXN],father[MAXN];
int top,stack[MAXN],size[MAXN],next[MAXN]/*必须要分开记录,不然中间会改变*/;
inline int getint(){
int w=0,q=0; char c=getchar(); while((c<'0'||c>'9') && c!='-') c=getchar();
if(c=='-') q=1,c=getchar(); while (c>='0'&&c<='9') w=w*10+c-'0',c=getchar(); return q?-w:w;
}
inline bool isroot(int x){ return (tr[father[x]][0]!=x) && (tr[father[x]][1]!=x); }
inline void update(int x){
int l=tr[x][0],r=tr[x][1]; size[x]=1;
if(l) size[x]+=size[l]; if(r) size[x]+=size[r];
}
inline void pushdown(int x){
if(tag[x]==0) return ;
int l=tr[x][0],r=tr[x][1];
tag[l]^=1; tag[r]^=1; swap(tr[x][0],tr[x][1]);
tag[x]=0;
}
inline void rotate(int x){
int y=father[x],z=father[y];
int l,r; l=(tr[y][1]==x); r=l^1;
if(!isroot(y)) tr[z][(tr[z][1]==y)]=x;
father[x]=z; father[y]=x;
tr[y][l]=tr[x][r]; father[tr[x][r]]=y; tr[x][r]=y;
update(y); update(x);
}
inline void splay(int x){
top=0; stack[++top]=x;
for(int i=x;!isroot(i);i=father[i]) stack[++top]=father[i];
for(int i=top;i>=1;i--) pushdown(stack[i])/*!!!不是top啊!!!*/;
int y,z;
while(!isroot(x)) {
y=father[x]; z=father[y];
if(!isroot(y)) {
if((tr[z][0]==y) ^ (tr[y][0]==x)) rotate(x);
else rotate(y);
}
rotate(x);
}
}
inline void access(int x){
int last=0;
while(x){
splay(x); tr[x][1]=last;
last=x; x=father[x];
}
}
inline void move_to_root(int x){
access(x);
splay(x);
tag[x]^=1;
}
inline void link(int x,int y){
move_to_root(x);
father[x]=y;
splay(x);
}
inline void cut(int x,int y){
move_to_root(x);
access(y);
splay(y);
tr[y][0]=father[x]=0;
}
inline void query(int x){
access(x);
splay(x);
printf("%d\n",size[tr[x][0]]);
}
inline void work(){
n=getint(); for(int i=1;i<=n;i++) val[i]=getint();
for(int i=1;i<=n;i++) {
if(i+val[i]>n) father[i]=n+1;
else father[i]=i+val[i];
next[i]=father[i];
//link(i,father[i]);
}
m=getint(); int type,x,y;
while(m--) {
type=getint();
if(type==1) {
move_to_root(n+1);
x=getint(); x++;
query(x);
}
else {
x=getint(); x++; y=getint();
cut(x,next[x]);
if(x+y>n) link(x,n+1),next[x]=n+1;
else link(x,x+y),next[x]=x+y;
}
}
}
int main()
{
work();
return 0;
}