洛谷P4216 [SCOI2015]情报传递（树剖+主席树）

传送门

 

我们可以进行离线处理，把每一个情报员的权值设为它开始收集情报的时间

那么设询问的时间为$t$，就是问路径上有多少个情报员的权值小于等于$t-c-1$

这个只要用主席树上树就可以解决了，顺便用树剖求一下LCA

//minamoto
#include<bits/stdc++.h>
using namespace std;
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
inline int read(){
    #define num ch-'0'
    char ch;bool flag=0;int res;
    while(!isdigit(ch=getc()))
    (ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getc());res=res*10+num);
    (flag)&&(res=-res);
    #undef num
    return res;
}
char sr[1<<21],z[20];int C=-1,Z;
inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
inline void print(int x,char ch){
    if(C>1<<20)Ot();
    while(z[++Z]=x%10+48,x/=10);
    while(sr[++C]=z[Z],--Z);sr[++C]=ch;
}
const int N=3e5+5,M=N*32;
struct node{int u,v,c,id;}q[N];
int head[N],Next[N],ver[N],tot;
inline void add(int u,int v){
    ver[++tot]=v,Next[tot]=head[u],head[u]=tot;
}
int L[M],R[M],sum[M],rt[N],cnt;
void update(int last,int &now,int l,int r,int x){
    sum[now=++cnt]=sum[last]+1;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(x<=mid) R[now]=R[last],update(L[last],L[now],l,mid,x);
    else L[now]=L[last],update(R[last],R[now],mid+1,r,x);
}
int fa[N],dep[N],top[N],sz[N],son[N],val[N],root,n,m,k;
void dfs1(int u){
    sz[u]=1,dep[u]=dep[fa[u]]+1;
    update(rt[fa[u]],rt[u],1,m,val[u]);
    for(int i=head[u];i;i=Next[i]){
        int v=ver[i];
        dfs1(v),sz[u]+=sz[v];
        if(sz[son[u]]<sz[v]) son[u]=v;
    }
}
void dfs2(int u,int t){
    top[u]=t;
    if(son[u]){
        dfs2(son[u],t);
        for(int i=head[u];i;i=Next[i])
        if(ver[i]!=son[u]) dfs2(ver[i],ver[i]);
    }
}
inline int LCA(int u,int v){
    while(top[u]!=top[v])
    dep[top[u]]>dep[top[v]]?u=fa[top[u]]:v=fa[top[v]];
    return dep[u]<dep[v]?u:v;
}
int query(int u,int v,int lca,int lca_fa,int l,int r,int x){
    if(r<=x) return sum[u]+sum[v]-sum[lca]-sum[lca_fa];
    if(l>x) return 0;
    int mid=(l+r)>>1,res=0;
    if(x>=l) res+=query(L[u],L[v],L[lca],L[lca_fa],l,mid,x);
    if(x>mid) res+=query(R[u],R[v],R[lca],R[lca_fa],mid+1,r,x);
    return res;
}
int main(){
//    freopen("testdata.in","r",stdin);
    n=read();
    for(int i=1;i<=n;++i){fa[i]=read(),add(fa[i],i);if(fa[i]==0) root=i;}
    m=read();for(int i=1;i<=n;++i) val[i]=m;
    for(int i=1;i<=m;++i){
        int op=read(),p;
        if(op&1) q[++k].u=read(),q[k].v=read(),q[k].c=read(),q[k].id=i;
        else val[p=read()]=i;
    }
    dfs1(root),dfs2(root,root);
    for(int i=1;i<=k;++i){
        int u=q[i].u,v=q[i].v,lca=LCA(u,v),lca_fa=fa[lca];
        print(dep[u]+dep[v]-2*dep[lca]+1,' ');
        int t=q[i].id-q[i].c-1;
        print(t>0?query(rt[u],rt[v],rt[lca],rt[lca_fa],1,m,t):0,'\n');
    }
    Ot();
    return 0;
}