bzoj4448 [Scoi2015]情报传递（树上主席树+离线）

我们把操作离线，先把每个点在什么时间被点亮记下来，然后直接建树上主席树即可。每次查询x-y的时间在1~Q-C-1的点有多少个。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(T==S){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,fa[N][20],rt[N],a[N],tot=0,dep[N],Log[N],owo=0;
vector<int>Son[N];
struct quer{
    int x,y,val;
}qq[N];
struct node{
    int ls,rs,x;
}tr[N*20];
inline void ins(int &p,int l,int r,int x){
    tr[++owo]=tr[p];p=owo;tr[p].x++;
    if(l==r) return;int mid=l+r>>1;
    if(x<=mid) ins(tr[p].ls,l,mid,x);
    else ins(tr[p].rs,mid+1,r,x);
}
inline void dfs(int x){
    for(int i=1;i<=Log[n];++i){
        if(!fa[x][i-1]) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }rt[x]=rt[fa[x][0]];if(a[x]) ins(rt[x],1,m,a[x]);
    for(int i=0;i<Son[x].size();++i){
        int y=Son[x][i];dep[y]=dep[x]+1;dfs(y);
    }
}
inline int ask(int p1,int p2,int p3,int p4,int l,int r,int x,int y){
    if(x<=l&&r<=y) return tr[p1].x+tr[p2].x-tr[p3].x-tr[p4].x;
    int mid=l+r>>1,res=0;
    if(x<=mid) res+=ask(tr[p1].ls,tr[p2].ls,tr[p3].ls,tr[p4].ls,l,mid,x,y);
    if(y>mid) res+=ask(tr[p1].rs,tr[p2].rs,tr[p3].rs,tr[p4].rs,mid+1,r,x,y);return res;
}
inline int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int d=dep[x]-dep[y];
    for(int i=0;i<=Log[d];++i)
        if(d>>i&1) x=fa[x][i];
    if(x==y) return x;
    for(int i=Log[n];i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();Log[0]=-1;
    for(int i=1;i<=n;++i) fa[i][0]=read(),Son[fa[i][0]].push_back(i),Log[i]=Log[i>>1]+1;
    m=read();
    for(int i=1;i<=m;++i){
        int op=read();
        if(op==2){int x=read();a[x]=i;continue;}
        qq[++tot].x=read();qq[tot].y=read();qq[tot].val=i-read()-1;
    }dfs(Son[0][0]);
    for(int i=1;i<=tot;++i){
        int x=qq[i].x,y=qq[i].y,t=lca(x,y);
        printf("%d ",dep[x]+dep[y]-2*dep[t]+1);
        if(qq[i].val<=0){puts("0");continue;}
        printf("%d\n",ask(rt[x],rt[y],rt[t],rt[fa[t][0]],1,m,1,qq[i].val));
    }return 0;
}