【bzoj4448】SCOI2015 情报传递

树剖，可以相当于一个染色问题，以及询问先前节点的染色个数。

把第二种操作离线掉，然后用线段树模拟即可。

#include<bits/stdc++.h>
#define N 800010
#define lson (o<<1)
#define rson (o<<1|1)
using namespace std;
struct Edge{int u,v,next;}G[N<<];
int head[N],tot=,wson[N],size[N],d[N],fa[N],top[N],tpos[N],pre[N],cnt,n;
struct Query{int id,t,x,y,opt;}a[N];
bool operator <(Query a,Query b){return (a.t<b.t)||(a.t==b.t&&a.opt==&&b.opt==);}
int sum[][N],rt;
inline int read(){
    int f=,x=;char ch;
    do{ch=getchar();if(ch=='-')f=-;}while(ch<''||ch>'');
    do{x=x*+ch-'';ch=getchar();}while(ch>=''&&ch<='');
    return f*x;
}
int f[N];
inline void addedge(int u,int v){
    G[++tot].u=u;G[tot].v=v;G[tot].next=head[u];head[u]=tot;
    G[++tot].u=v;G[tot].v=u;G[tot].next=head[v];head[v]=tot;
}
void dfs1(int u,int f){
    size[u]=;
    for(int i=head[u];i;i=G[i].next){
        int v=G[i].v;if(v==f)continue;
        fa[v]=u;d[v]=d[u]+;
        dfs1(v,u);
        size[u]+=size[v];
        if(size[v]>size[wson[u]])wson[u]=v;
    }
}
void dfs2(int u,int tp){
    tpos[u]=++cnt;pre[cnt]=u;top[u]=tp;
    if(wson[u])dfs2(wson[u],tp);
    for(int i=head[u];i;i=G[i].next){
        int v=G[i].v;if(v==fa[u]||v==wson[u])continue;
        dfs2(v,v);
    }
}
struct Segment_Tree{
    int cntv[N<<];
    void change(int o,int l,int r,int q){
        cntv[o]++;if(l==r)return;
        int mid=(l+r)>>;
        if(q<=mid)change(lson,l,mid,q);
        else change(rson,mid+,r,q);
    }
    int query(int o,int l,int r,int ql,int qr){
        if(ql<=l&&r<=qr)return cntv[o];
        int mid=(l+r)>>,ans=;
        if(ql<=mid)ans+=query(lson,l,mid,ql,qr);
        if(qr>mid)ans+=query(rson,mid+,r,ql,qr);
        return ans;
    }
}T;
void query(int u,int v,int id){
    int ans=;int xx=u,yy=v;f[id]=;
    while(top[u]!=top[v]){
        if(d[top[u]]<d[top[v]])swap(u,v);
        ans+=T.query(,,n,tpos[top[u]],tpos[u]);u=fa[top[u]];
    }
    if(d[u]<d[v])swap(u,v);
    ans+=T.query(,,n,tpos[v],tpos[u]);
    sum[][id]=d[xx]+d[yy]-*d[v]+;sum[][id]=ans;
}
int main(){
    n=read();
    for(int i=;i<=n;i++){fa[i]=read();if(!fa[i])rt=i;else addedge(fa[i],i);}
    dfs1(rt,);dfs2(rt,rt);int m=read();
    for(int i=;i<=m;i++){
        a[i].opt=read();a[i].id=i;
        if(a[i].opt==){a[i].x=read();a[i].t=i;}
        else{
            a[i].x=read();a[i].y=read();int tmp=read();
            a[i].t=i-tmp-;
        }
    }
    sort(a+,a+m+);
    for(int i=;i<=m;i++)if(a[i].opt==)T.change(,,n,tpos[a[i].x]);
    else query(a[i].x,a[i].y,a[i].id);
    for(int i=;i<=m;i++)if(f[i])printf("%d %d\n",sum[][i],sum[][i]);
}