hdu6394 Tree dfs序+分块

题目连接：Tree

题意：跟弹飞绵羊差不多这题是在树上条，给你一颗树，每个点上有个值w，当你走到这会直接跳到向上走w步的父亲节点（如过没有就算跳出去了）然后两种操作。

一种询问从x跳出要多少步。第二种修改某个点的w.

题解，先dfs一次的到dfs序，然后按dfs序分块，有cn保存它跳出这一块需要的次数，net保存跳出这块会去的点。然后就都是暴力了。复杂度nsqrt(n);

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5+10;
int w[N];//
int fa[20][N];
int n,m;
int head[N],nxt[N],to[N],cnt;//链表存边,cnt边数
int id[N],tot;//dfs序
int num,block,belong[N],l[N],r[N];//块的数目,快的大小，属于那块，每块的左右边界
int cn[N],net[N],pre[N];//跳出本块的次数,跳向的下一块的点,这个点会跳向的位置
void init()//初始化
{
    memset(head,0,sizeof(head));
    cnt=0;
    tot=0;
}
void add_edge(int v,int u)//加v到u的单向边
{
    to[++cnt]=u;nxt[cnt]=head[v]; head[v]=cnt;
}
void dfs(int v,int f)//dfs一次,得到dfs序和每一个点的父亲
{
    fa[0][v]=f;id[v]=++tot;
    for(int i=head[v];i;i=nxt[i])
    {
        dfs(to[i],v);
    }
}
void built()//分块
{
    block=sqrt(n);
    num=n/block;if(n%block)num++;
    for(int i=1;i<=n;i++)
    {
        belong[i]=((i-1)/block)+1;
    }
    for(int i=1;i<=num;i++)
    {
        l[i]=(i-1)*block+1;r[i]=i*block;
    }
    r[num]=n;
}
int find(int x,int l)//x向上走l步会到那里
{
    int p=x;
    for(int i=19;i>=0;i--)
    {
        if((l>>i)&1)
        {
            x=fa[i][x];
        }
    }
    return x;
}
void dfs2(int v)//再次dfs得到cn[N],net[N],pre[N]
{
    int p=find(v,w[v]);
    pre[id[v]]=id[p];
    if(id[p]<l[belong[id[v]]])cn[id[v]]=1,net[id[v]]=id[p];
    else cn[id[v]]=cn[id[p]]+1,net[id[v]]=net[id[p]];
     for(int i=head[v];i;i=nxt[i])
    {
        dfs2(to[i]);
    }
}
int query(int x)
{
    int ans=0;
    while(x>0)
    {
        ans+=cn[x];
        x=net[x];
    }
    return ans;
}
void update(int x,int v)
{
    int p=find(x,v);w[x]=v;
    pre[id[x]]=id[p];
    if(id[p]<l[belong[id[x]]])cn[id[x]]=1,net[id[x]]=id[p];
    else cn[id[x]]=cn[id[p]]+1,net[id[x]]=net[id[p]];
    for(int i=id[x]+1;i<=r[belong[id[x]]];i++)//更新这块内后面的点
    {
        if(pre[i]>=l[belong[i]])
        {
            cn[i]=cn[pre[i]]+1;
            net[i]=net[pre[i]];
        }
    }
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d",&n);
        for(int i=2;i<=n;i++)
        {
            int x;
            scanf("%d",&x);
            add_edge(x,i);
        }
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&w[i]);
        }
        dfs(1,0);
        for(int i=1;i<20;i++)
        {
            for(int j=1;j<=n;j++)
            {
                fa[i][j]=fa[i-1][fa[i-1][j]];
            }
        }
        built();
        dfs2(1);
        scanf("%d",&m);
        while(m--)
        {
            int op,x,v;
            scanf("%d %d",&op,&x);
            if(op==1)
            {
                printf("%d
",query(id[x]));
            }
            else
            {
                scanf("%d",&v);
                update(x,v);
            }
        }
    }
}