【题目链接】 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; }
时间: 2024-10-10 06:57:48