题目链接:https://codeforces.com/problemset/problem/208/E
题目大意:
给你一个有根树,树的根为0节点,给你m个询问,每次询问会询问,对于v这个节点的第p个祖先vp,有多少个节点的第p个祖先也是vp?如果是0那就输出0。
思路:
这题让我理解了如果求树上共享祖先的同一深度的节点数
首先,我们必须得找到一个节点vp,它是v的第p个祖先,那么这个操作可以用倍增完成,道理和LCA是一样的,这里给出模板代码:
int p[maxn][20];//表示点i的第2^i个祖先是哪个
int d[maxn];//深度
void dfs(int u,int fa){
d[u]=d[fa]+1;
p[u][0]=fa;
for(int i=1;i<POW;i++) p[u][i]=p[p[u][i-1]][i-1];
int sz=G[u].size();
for(int i=0;i<sz;i++){
int v=G[u][i];
if(v==fa) continue;
dfs(v,u);
}
}
//求a和b的LCA
int lca( int a, int b ){
if( d[a] > d[b] ) a ^= b, b ^= a, a ^= b;
if( d[a] < d[b] ){
int del = d[b] - d[a];
for( int i = 0; i < POW; i++ ) if(del&(1<<i)) b=p[b][i];
}
if( a != b ){
for( int i = POW-1; i >= 0; i-- )
if( p[a][i] != p[b][i] )
a = p[a][i] , b = p[b][i];
a = p[a][0], b = p[b][0];
}
return a;
}
//寻找v的第pp个祖先
int Find(int v,int pp){
for(int i=0;i<18;i++){
if(pp&(1<<i))v=p[v][i];
}
return v;
}
用这个Find函数就可找到父节点了。
然后,我们对每个节点,记录一下这个节点的三个信息:节点深度depth[u],节点入度dfs序in[u],节点出度dfs序out[u]
我们设它的第p的父节点为vp
我们用一个vector记录每个深度的入度和出度,然后对于每个v,我们lower_bound查找相同深度的第一个大于等于它的父节点vp的入度的位置,再用lower_bound查找相同深度第一个大于等于它的父节点vp的出度的位置,两者形成一个区间,这个区间的长度,就是vp的子树的大小,然后你是在同一个深度查找的,所以现在的长度,就是vp的子树且和v在同一个深度的节点数个数
#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
vector<int>G[maxn];
vector<int>g[maxn];
void init(int n){
for(int i=0;i<=n;i++){
G[i].clear();
g[i].clear();
}
}
//用于求解LCA和保存公共祖先的dfs模板
int p[maxn][20];//表示点i的第2^i个祖先是哪个
int in[maxn];
int out[maxn];
int dep[maxn];
int dfn,depth;
const int POW =18;
int vis[maxn];
void dfs(int u,int fa){
in[u]=++dfn;
vis[u]=1;
dep[u]=dep[fa]+1;
depth=max(depth,dep[u]);
g[dep[u]].push_back(dfn);//记录同一个深度的dfn
p[u][0]=fa;//u的第一个父亲为fa
for(int i=1;i<18;i++)p[u][i]=p[p[u][i-1]][i-1];//寻找第2^i的父亲
for(int i=0;i<G[u].size();i++){
int v=G[u][i];
if(vis[v])continue;
dfs(v,u);
}
out[u]=++dfn;//出的dfn
}
//寻找v的第pp个祖先
int Find(int v,int pp){
for(int i=0;i<18;i++){
if(pp&(1<<i))v=p[v][i];
}
return v;
}
int ans[maxn];
signed main(){
int n;
while(scanf("%d",&n)!=EOF){
init(n);
dfn=0;//dfs序
memset(p,0,sizeof(p));
dep[0]=0;
depth=0;
for(int i=1;i<=n;i++){
int u;
scanf("%d",&u);
G[u].push_back(i);
G[i].push_back(u);
}
int root=0;
memset(vis,0,sizeof(vis));
dfs(0,0);
int T;
for(int i=0;i<=n;i++){
cout<<i<<" in "<<in[i]<<endl;
cout<<i<<" out "<<out[i]<<endl;
}
scanf("%d",&T);
int x=T;
int inx=0;
while(T--){
int v,pp;
scanf("%d%d",&v,&pp);
int vp=Find(v,pp);//祖先
//cout<<"debug "<<vp<<endl;
if(vp==0){
ans[++inx]=0;
}
else{
int x1=lower_bound(g[dep[v]].begin(),g[dep[v]].end(),in[vp])-g[dep[v]].begin();
int y1=lower_bound(g[dep[v]].begin(),g[dep[v]].end(),out[vp])-g[dep[v]].begin();
ans[++inx]=y1-x1-1;
}
}
for(int i=1;i<=inx;i++)printf("%d ",ans[i]);
puts("");
}
}
/*
0 1 2 2 1 5 5 0 8 9 9 8 11 11
*/