【CF208E】Blood Cousins

题目大意：给定一个 N 个点的森林，M 个询问，每次询问对于点 u 来说，有多少个点和 u 有相同的 K 级祖先。

题解：线段树合并适合处理子树贡献的问题。
发现要回答这个询问在点 u 处计算很困难，但是在 u 的 k 级祖先处处理询问很简单，即：问对于 v 子树中深度为 k 的节点的个数。因此，采用将询问离线，对于每个点的询问转化到其祖先节点，最后采用线段树合并即可。

代码如下

#include <bits/stdc++.h>
#define mp make_pair
#define pb push_back
using namespace std;
const int maxn=1e5+10;

int n,m,ans[maxn];
vector<int> G[maxn];
int f[maxn][21],dep[maxn];
vector<pair<int,int>> q[maxn]; // dep, id

void pre(int u,int fa){
    dep[u]=dep[fa]+1,f[u][0]=fa;
    for(int i=1;i<=20;i++)f[u][i]=f[f[u][i-1]][i-1];
    for(auto v:G[u])pre(v,u);
}
int LA(int x,int k){
    int t=dep[x]-k;
    for(int i=20;~i;i--)if(dep[f[x][i]]>=t)x=f[x][i];
    return x;
}

struct node{
    #define ls(o) t[o].lc
    #define rs(o) t[o].rc
    int lc,rc,sz;
}t[maxn*20];
int tot,root[maxn];
inline void pushup(int o){
    t[o].sz=t[ls(o)].sz+t[rs(o)].sz;
}
void insert(int &o,int l,int r,int pos){
    if(!o)o=++tot;
    if(l==r){++t[o].sz;return;}
    int mid=l+r>>1;
    if(pos<=mid)insert(ls(o),l,mid,pos);
    else insert(rs(o),mid+1,r,pos);
    pushup(o);
}
int merge(int x,int y,int l,int r){
    if(!x||!y)return x+y;
    if(l==r){t[x].sz+=t[y].sz;return x;}
    int mid=l+r>>1;
    ls(x)=merge(ls(x),ls(y),l,mid);
    rs(x)=merge(rs(x),rs(y),mid+1,r);
    return pushup(x),x;
}
int query(int o,int l,int r,int pos){
    if(!o)return 0;
    if(l==r)return t[o].sz;
    int mid=l+r>>1;
    if(pos<=mid)return query(ls(o),l,mid,pos);
    else return query(rs(o),mid+1,r,pos);
}

void dfs(int u){
    for(auto v:G[u]){
        dfs(v);
        root[u]=merge(root[u],root[v],1,n);
    }
    for(auto p:q[u]){
        int d=p.first,id=p.second;
        ans[id]=query(root[u],1,n,d)-1;
    }
    insert(root[u],1,n,dep[u]);
}
void read_and_parse(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&f[i][0]);
        if(!f[i][0])continue;
        G[f[i][0]].pb(i);
    }
    for(int i=1;i<=n;i++)if(!f[i][0])pre(i,0);
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        int v,p;
        scanf("%d%d",&v,&p);
        int u=LA(v,p);
        if(!u)continue;
        q[u].pb(mp(dep[v],i));
    }
}
void solve(){
    for(int i=1;i<=n;i++)if(!f[i][0])dfs(i);
    for(int i=1;i<=m;i++)printf("%d ",ans[i]);
}
int main(){
    read_and_parse();
    solve();
    return 0;
} 

转载于:https://www.cnblogs.com/wzj-xhjbk/p/10985953.html