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的情况,你都要输出一个需要的步数,占一行。
LCT模板。
#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define Pil pair<int,ll>
#define Pli pair<ll,int>
#define Pll pair<ll,ll>
#define pb push_back
#define pc putchar
#define mp make_pair
#define file(k) memset(k,0,sizeof(k))
#define ll long long
#define ls ch[x][1]
#define rs ch[x][0]
int rd()
{
int sum = 0;char c = getchar();bool flag = true;
while(c < '0' || c > '9') {if(c == '-') flag = false;c = getchar();}
while(c >= '0' && c <= '9') sum = sum * 10 + c - 48,c = getchar();
if(flag) return sum;
else return -sum;
}
const int maxn = 2010000;
struct tree
{
int ch[maxn][2],fa[maxn],v[maxn],mx[maxn],rev[maxn],sum[maxn];
int q[maxn],top;
void update(int x)
{
sum[x] = sum[ls] + sum[rs] + v[x];
mx[x] = v[x];
if(ls) mx[x] = max(mx[x],mx[ls]);
if(rs) mx[x] = max(mx[x],mx[rs]);
}
bool isroot(int x){return ch[fa[x]][1] != x && ch[fa[x]][0] != x;}
void pushdown(int x){if(rev[x])rev[x] ^= 1,swap(ls,rs),rev[ls] ^= 1,rev[rs] ^= 1;}
void rorate(int x)
{
int y = fa[x],z = fa[y],k = ch[y][1] == x;
if(!isroot(y)) ch[z][ch[z][1] == y] = x;
ch[y][k] = ch[x][k^1]; fa[ch[x][k^1]] = y;
ch[x][k^1] = y; fa[x] = z;fa[y] = x;
update(y);update(x);
}
void splay(int x)
{
int k = x;top = 0;
while(1)
{
q[++top] = k;
if(isroot(k)) break;
k = fa[k];
}
while(top) pushdown(q[top--]);
while(!isroot(x))
{
int y = fa[x],z = fa[y];
if(!isroot(y)) rorate((ch[z][1]==y)^(ch[y][1] == x)?y:x);
rorate(x);
}
}
void access(int x){for(int t=0;x;x = fa[t=x]) splay(x),ch[x][1]=t,update(x);}
void makeroot(int x){access(x);splay(x);rev[x] ^= 1;}
void split(int x,int y){makeroot(x);access(y);splay(y);}
void link(int x,int y){makeroot(x);fa[x] = y;}
void cut(int x,int y)
{
makeroot(x);access(y);splay(y);
fa[x] = ch[y][0] = 0;update(y);
}
}T;
int n,m;
int a[201000];
int main()
{
n = rd();
rep(i,1,n)
{
T.v[i] = 1;
a[i] = rd();
if(a[i] + i > n) a[i] = n+1-i;
T.link(i,a[i]+i);
}
m = rd();
rep(i,1,m)
{
int k = rd();
if(k == 1)
{
int x = rd()+1;
T.split(x,n+1);
printf("%d\n",T.sum[n+1]);
}
else
{
int x = rd()+1,v = rd();
if(x + v > n) v = n+1-x;
T.cut(x,x+a[x]);
a[x] = v;
T.link(x,x+a[x]);
}
}
return 0;
}