【CodeForces 208E】Blood Cousins dps序+二分

开一个vector记录所有深度的dfs序，每一次查询的时候先倍增到他的祖先找到进出时间戳然后再这一个点的深度进行二分查找这个给出的时间戳就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
#include<algorithm>
#define maxn 100010
using namespace std;
int n,f[maxn][20],head[maxn],tot,tim,Max;
struct edge{int v,next;}e[maxn*2];
void adde(int a,int b){e[tot].v=b,e[tot].next=head[a];head[a]=tot++;}
struct node{
	int h,in,out;
	bool operator <( const node& b)const{
		if(h!=b.h)return h<b.h;
		return in==b.in? in<b.in : out<b.out;
	}
}nod[maxn];
vector<int >g[maxn];

void dfs(int u,int fa){
	f[u][0]=fa;nod[u].in=++tim;
	for(int i=1;i<=17;i++)f[u][i]=f[f[u][i-1]][i-1];
	nod[u].h=nod[fa].h+1;Max=max(Max,nod[u].h);
	g[nod[u].h].push_back(nod[u].in);
	for(int i=head[u];i!=-1;i=e[i].next){
		int v=e[i].v;if(v==fa)continue;
		dfs(v,u);
	}
	nod[u].out=++tim;
}

void solve(int u,int p){
	int now=u,last=p;
	for(int i=17;i>=0;i--){
		int to=f[now][i];
		if((nod[now].h-nod[to].h>last))continue;
		last-=(nod[now].h-nod[to].h);
		now=f[now][i];
	}
	if(now==0){
		printf("0 ");return;
	}
	int a=nod[now].in,b=nod[now].out,h=nod[u].h;
	int l=lower_bound(g[h].begin(),g[h].end(),a)-g[h].begin();
	int r=upper_bound(g[h].begin(),g[h].end(),b)-g[h].begin()-1;
	printf("%d ",r-l);
}

int main(){
	memset(head,-1,sizeof(head));
	scanf("%d",&n);
	for(int x,i=1;i<=n;i++){
		scanf("%d",&x);
		adde(i,x),adde(x,i);
	}
	dfs(0,0);int T,a,b;
	for(int i=1;i<=Max;i++)sort(g[i].begin(),g[i].end());
	scanf("%d",&T);
	while(T--){
		scanf("%d%d",&a,&b);
		solve(a,b);
	}
	return 0;
}