CodeForces - 208E Blood Cousins 第k祖先+dfs序+二分

题目链接：点击查看

题意：n个定点，给出每个点的父亲，0代表没有父亲，可能是个森林，m次询问，v,p表示和v同深度的，第p个祖先一样的点还用几个

题解：第几个祖先我们可以用倍增来解决，祖先相同同深度怎么解决呢？我们可以用dfs序来解决，把每个深度的用vector存一下，因为我们dfs遍历的时候同一层的是连续储存的，所以我们就可以二分查找了

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
vector<int> v[N];
vector<int> dep[N];
int in[N],out[N],cnt;
int dp[N][20];
int ind[N];
int indegree[N];
int n,m;
void dfs1(int u,int deep)
{
	int to;
	in[u]=++cnt;
	ind[u]=deep;
	for(int i=0;i<20;i++)
		if(dp[u][i])
			dp[u][i+1]=dp[dp[u][i]][i];
		else
			break;
	dep[deep].push_back(cnt);
// 	cout<<u<<endl;
	for(int i=0;i<v[u].size();i++)
	{
		to=v[u][i];
		dp[to][0]=u;
		dfs1(to,deep+1);
	}
	out[u]=cnt;
}
int getfath(int x,int up)
{
	for(int i=0;i<20;i++)
		if(up & (1<<i))
			x=dp[x][i];
	return x;
}
int main()
{
	int x,up,u;
	int l,r;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&x);
		if(x) v[x].push_back(i),indegree[i]++;
	}
	for(int i=1;i<=n;i++)
	{
		if(indegree[i]==0)
			dfs1(i,1);
	}
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&up);
		u=getfath(x,up);
//		cout<<u<<endl;
		if(!u) printf("0\n");
		else
		{
			r=upper_bound(dep[ind[x]].begin(),dep[ind[x]].end(),out[u])-dep[ind[x]].begin();
			l=lower_bound(dep[ind[x]].begin(),dep[ind[x]].end(),in[u])-dep[ind[x]].begin();
			printf("%d\n",r-l-1);
		}
	}
	return 0;
}