题目:某天,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的情况,你都要输出一个需要的步数,占一行。
思路:看成n个点,再加一个编号为n+1的点,跳出的点pre指向n+1,这样就成了一棵树,然后就可以套LCT了
代码:
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<list>
#include<numeric>
using namespace std;
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f
// 0x3f3f3f3f
const int maxn=2e5+50;
const int maxe=2*maxn;
struct Edge{
int to,next;
}edge[maxe];
int ch[maxn][2],key[maxn],pre[maxn],rev[maxn];
int siz[maxn];
bool rt[maxn];//标记节点是不是splay的根
int n;//n个节点,从1开始计数
void init(){
for(int i=0;i<=n+1;i++){
rt[i]=true;
ch[i][0]=ch[i][1]=0;
pre[i]=0;
rev[i]=0;
siz[i]=1;
}
siz[0]=0;
}
void update_rev(int x){
if(!x) return;
swap(ch[x][0],ch[x][1]);
rev[x]^=1;
}
void push_down(int x){
if(rev[x]){
update_rev(ch[x][0]);
update_rev(ch[x][1]);
rev[x]=0;
}
}
void push_up(int x){
siz[x]=1;
siz[x]+=siz[ch[x][0]];
siz[x]+=siz[ch[x][1]];
}
void rotate(int x){
int y=pre[x],d=ch[y][1]==x;
ch[y][d]=ch[x][!d];
pre[ch[y][d]]=y;
pre[x]=pre[y];
pre[y]=x;
ch[x][!d]=y;
if(rt[y]) rt[y]=false,rt[x]=true;
else ch[pre[x]][ch[pre[x]][1]==y]=x;
push_up(y);
}
//P函数先将根结点到x的路径上所有的结点的标记逐级下放
void P(int x){
if(!rt[x]) P(pre[x]);
push_down(x);
}
//将x旋转到根
void splay(int x){
P(x);
while(!rt[x]){
int f=pre[x],ff=pre[f];
if(rt[f]) rotate(x);
else if((ch[ff][1]==f)==(ch[f][1]==x))
rotate(f),rotate(x);
else
rotate(x),rotate(x);
}
push_up(x);
}
//将x到根的路径变成首选边
int Access(int x){
int y=0;
for(;x;x=pre[y=x]){
splay(x);
rt[ch[x][1]]=true;
rt[ch[x][1]=y]=false;
push_up(x);
}
return y;
}
int arr[maxn];
int main(){
int m,op,u,v;
while(~scanf("%d",&n)){
init();
for(int i=1;i<=n;i++)
scanf("%d",&arr[i]);
for(int i=1;i<=n;i++){
int t=i+arr[i];
if(t>n) t=n+1;
pre[i]=t;
}
scanf("%d",&m);
while(m--){
scanf("%d",&op);
if(op==1){
scanf("%d",&u);u++;
Access(u);
splay(u);
printf("%d\n",siz[ch[u][0]]);
}
else{
scanf("%d%d",&u,&v);u++;
Access(u);
splay(u);
pre[ch[u][0]]=0;
rt[ch[u][0]]=true;
ch[u][0]=0;
int t=u+v;
if(t>n) t=n+1;
pre[u]=t;
}
}
}
return 0;
}