BZOJ 2002 [Hnoi2010]Bounce 弹飞绵羊（分块）

 

【题目链接】 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;
}