CF208E 題解

zxh923發表於2024-07-18

Blood Cousins

前置知識:線段樹合併。

我們先把題目轉化一下。這裡先設 \(v\)\(p\) 級祖先為 \(u\),事實上要求的東西就是 \(u\)\(p\) 級後代的個數減 \(1\),減 \(1\) 是因為要把自己減去。顯然這個目標點 \(t\) 要滿足兩個要求:

  • \(t\)\(u\) 子樹內。

  • \(dep_u\) 表示 \(u\) 的深度,則 \(dep_u+p=dep_t\)

於是我們對每個點建一棵權值線段樹,維護深度,然後把詢問離線下來,如果這個點有詢問,我們就在合併後的權值線段樹內查詢目標深度的點的個數即可。

然後就說完了,直接看一下程式碼:

#include<bits/stdc++.h>
#define int long long
#define N 100005
#define M 200005
#define K 23
#define pii pair<int,int>
#define x first
#define y second
using namespace std;
int n,m,tot;
int h[N],e[M],ne[M],idx;
int fa[N][K],dep[N],res[N],rt[N];
vector<int>root;
vector<pii>ask[N];
void add(int a,int b){
	e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}
void dfs1(int u,int f){
	dep[u]=dep[f]+1;
	fa[u][0]=f;
	for(int i=1;i<K;i++){
		fa[u][i]=fa[fa[u][i-1]][i-1];
	}
	for(int i=h[u];~i;i=ne[i]){
		int j=e[i];
		if(j==f)continue;
		dfs1(j,u);
	}
}
int get(int u,int k){
	for(int i=K-1;~i;i--){
		if(k>>i&1){
			u=fa[u][i];
		}
	}
	return u;
}
struct node{
	int l,r,sum;
}tr[N<<5];
void pushup(int u){
	tr[u].sum=tr[tr[u].l].sum+tr[tr[u].r].sum;
}
void modify(int &u,int l,int r,int p){
	if(!u)u=++tot;
	if(l==r){
		tr[u].sum++;
		return;
	}
	int mid=l+r>>1;
	if(p<=mid)modify(tr[u].l,l,mid,p);
	else modify(tr[u].r,mid+1,r,p);
	pushup(u);
}
int merge(int x,int y,int l,int r){
	if(!x||!y)return x+y;
	if(l==r){
		tr[x].sum+=tr[y].sum;
		return x;
	}
	int mid=l+r>>1;
	tr[x].l=merge(tr[x].l,tr[y].l,l,mid);
	tr[x].r=merge(tr[x].r,tr[y].r,mid+1,r);
	pushup(x);
	return x;
}
int qry(int u,int l,int r,int p){
	if(l==r)return tr[u].sum;
	int mid=l+r>>1;
	if(p<=mid)return qry(tr[u].l,l,mid,p);
	else return qry(tr[u].r,mid+1,r,p);
}
void dfs2(int u){
	modify(rt[u],1,n,dep[u]);
	for(int i=h[u];~i;i=ne[i]){
		int j=e[i];
		if(j==fa[u][0])continue;
		dfs2(j);
		rt[u]=merge(rt[u],rt[j],1,n);
	}
	for(auto eu:ask[u]){
		res[eu.y]=qry(rt[u],1,n,eu.x)-1;
	}
}
signed main(){
	cin>>n;
	memset(h,-1,sizeof h);
	for(int i=1;i<=n;i++){
		int x;
		cin>>x;
		if(x)add(x,i),add(i,x);
		else root.push_back(i);
	}
	for(auto i:root){
		dep[i]=1;
		dfs1(i,0);
	}
	cin>>m;
	for(int i=1;i<=m;i++){
		int x,k;
		cin>>x>>k;
		int p=get(x,k);
		if(p==0){
			res[i]=0;
			continue;
		}
		ask[p].push_back({k+dep[p],i});
	}
	for(auto i:root){
		dfs2(i);
	}
	for(int i=1;i<=m;i++){
		cout<<res[i]<<' ';
	}
	return 0;
}