【题目链接】 http://www.lydsy.com/JudgeOnline/problem.php?id=2002
【题目大意】
给出一片森林,操作允许更改一个节点的父亲,查询一个节点的深度。
父亲节点的编号一定大于子节点
【题解】
我们将所有的节点按照序号分块,记录其到下一个分块的深度,
以及其可以跳到下一个分块的位置,
那么修改节点的父亲对于所存储的数据的影响就被缩小到一个块内,单次修改sqrt(n)。
动态树做法:链接
【代码】
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
const int N=200010;
int n,m,op,pos[N],block,l[N],r[N],K[N],d[N],nxt[N];
int cal(int x){for(int res=0;;x=nxt[x]){res+=d[x];if(!nxt[x])return res;}}
int main(){
scanf("%d",&n);
block=sqrt(n+0.5);
for(int i=1;i<=n;i++)scanf("%d",&K[i]);
if(n%block)m=n/block+1;
else m=n/block;
for(int i=1;i<=m;i++)l[i]=(i-1)*block+1,r[i]=i*block; r[m]=n;
for(int i=1;i<=n;i++)pos[i]=(i-1)/block+1;
for(int i=n;i;i--){
if(i+K[i]>n)d[i]=1;
else if(pos[i]==pos[i+K[i]])d[i]=d[i+K[i]]+1,nxt[i]=nxt[i+K[i]];
else d[i]=1,nxt[i]=i+K[i];
}scanf("%d",&op);
while(op--){
int u,x,y;
scanf("%d%d",&u,&x);x++;
if(u==1)printf("%d\n",cal(x));
else{
scanf("%d",&y); K[x]=y;
for(int i=x;i>=l[pos[x]];i--)if(pos[i]==pos[i+K[i]]){
d[i]=d[i+K[i]]+1; nxt[i]=nxt[i+K[i]];
}else d[i]=1,nxt[i]=i+K[i];
}
}return 0;
}