这道题不用换根,因为换根存在的意义就是方便查询路径(u,v)上的信息,而这道题只需要一个点的size
#include<bits/stdc++.h>
const int N=200005;
using namespace std;
template<class T>
inline void read(T &x)
{
x=0; int f=1;
static char ch=getchar();
while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
x*=f;
}
int n,d[N],m;
struct Link_Cut_Tree
{
int lc,rc,size,fa;
}t[N];
inline void pushup(int u){t[u].size=t[t[u].lc].size+t[t[u].rc].size+1;}
inline int which(int u){return t[t[u].fa].rc==u;}
inline bool isroot(int u){return t[t[u].fa].lc!=u&&t[t[u].fa].rc!=u;}
inline void Rotate(int u)
{
int v=t[u].fa,w=t[v].fa;
int b=(t[v].rc==u)?t[u].lc:t[u].rc;
if(!isroot(v)) (t[w].lc==v?t[w].lc:t[w].rc)=u;
t[u].fa=w; t[v].fa=u;
if(b) t[b].fa=v;
if(t[v].rc==u) t[v].rc=b,t[u].lc=v;
else t[v].lc=b,t[u].rc=v;
pushup(v); pushup(u);
}
inline void Splay(int u)
{
while(!isroot(u))
{
if(!isroot(t[u].fa))
{
if(which(u)==which(t[u].fa)) Rotate(t[u].fa);
else Rotate(u);
}
Rotate(u);
}
}
inline void access(int u)
{
for(int v=0;u;u=t[v=u].fa)
Splay(u),t[u].rc=v,pushup(u);
}
inline void Link(int u,int v){ Splay(u); t[u].fa=v;} //刚刚access过了
inline void Cut(int u)
{
access(u); Splay(u);
t[t[u].lc].fa=0; t[u].lc=0; pushup(u);
}
inline void write(int x)
{
if(x<0){x=-x; write(x);}
if(x>9) write(x/10);
putchar(x%10+'0');
}
int main()
{
read(n);
for(int i=1;i<=n;i++)
{
read(d[i]); t[i].size=1;
if(i+d[i]<=n) Link(i,i+d[i]);
}
read(m);
for(int i=1,opt,x,y;i<=m;i++)
{
read(opt); read(x); x++;
if(opt==1)
{
access(x); Splay(x);
write(t[x].size),putchar('\n');
}
else
{
read(y); Cut(x); d[x]=y;
if(x+d[x]<=n) Link(x,x+d[x]);
}
}
return 0;
}