bzoj 3631 松鼠的新家

挺难得有没有。。

查了查题解

没有看太明白区间++–，遂放弃

决定使用树剖+线段树+lazy标记。

嗯所以代码量有点可观。。。。。

#include <cstdio>  
#include <cstring>  
#include <iostream>  
#include <algorithm>  
using namespace std;  
#define lson pos<<1  
#define rson pos<<1|1  
#define N 300005  
int n,a[N],ans[N];  
int cnt,head[N],next[2*N],to[2*N];  
int now,size[N],deep[N],fa[N],son[N],p_id[N],id_p[N],top[N];  
int f[4*N];  
void add(int x,int y)  
{  
    to[++cnt]=y;  
    next[cnt]=head[x];  
    head[x]=cnt;  
}  
void dfs(int x,int d)  
{  
    size[x]=1;deep[x]=d;  
    for(int y,i=head[x];i;i=next[i])  
        if((y=to[i])!=fa[x])  
        {  
            fa[y]=x;  
            dfs(y,d+1);  
            if(size[y]>size[son[x]])  
                son[x]=y;  
            size[x]+=size[y];  
        }  
}  
void create(int x,int d)  
{  
    p_id[x]=++now;id_p[now]=x;  
    id_p[now]=x;  
    top[x]=d;  
    if(son[x])  
        create(son[x],d);  
    for(int y,i=head[x];i;i=next[i])  
        if((y=to[i])!=fa[x]&&y!=son[x])  
            create(y,y);  
}  
void lazy(int pos)  
{  
    f[lson]+=f[pos];  
    f[rson]+=f[pos];  
    f[pos]=0;  
}  
void fix(int pos,int l,int r,int x,int y)  
{  
    if(x<=l&&r<=y){ f[pos]++; return; }  
    if(f[pos]) lazy(pos);  
    int mid=(l+r)>>1;  
    if(y<=mid)  
        fix(lson,l,mid,x,y);  
    else if(x>mid)  
        fix(rson,mid+1,r,x,y);  
    else 
        fix(lson,l,mid,x,y),  
        fix(rson,mid+1,r,x,y);  
}  
void getans(int pos,int l,int r)  
{  
    if(l==r){ ans[id_p[l]]=f[pos]; return; }  
    if(f[pos]) lazy(pos);  
    int mid=(l+r)>>1;  
    getans(lson,l,mid);  
    getans(rson,mid+1,r);  
}  
void work(int x,int y)  
{  
    int f1=top[x],f2=top[y];  
    while(f1!=f2)  
    {  
        if(deep[f1]<deep[f2])  
            swap(x,y),swap(f1,f2);  
        fix(1,1,n,p_id[f1],p_id[x]);  
        x=fa[f1];  
        f1=top[x];  
    }  
    if(deep[x]>deep[y])  
        swap(x,y);  
    fix(1,1,n,p_id[x],p_id[y]);  
}  
int main()  
{  
    cin>>n;  
    for(int i=1;i<=n;i++)  
        scanf("%d",&a[i]);  
    for(int x,y,i=1;i<n;i++)  
        scanf("%d%d",&x,&y),  
        add(x,y),add(y,x);  
    dfs(1,1);  
    create(1,1);  
    for(int i=2;i<=n;i++)  
        work(a[i],a[i-1]);  
    getans(1,1,n);  
    for(int i=1;i<=n;i++)  
        printf("%d\n",i==a[1]?ans[i]:ans[i]-1);  
    return 0;  
}