题意:
就是给你一个家族树,不一定就一个树,可能是森林。然后如果a是b的父亲节点,那么a就是b的一级祖先,依次就是k级祖先。如果两个点共同有k级祖先,那么他们就是k级表亲。现在给你m个询问,问你和a点为k级表亲的点有多少个。
思考:
刚开始看到,这不是子树问题啊,这是找相邻的兄弟问题啊。但是你想想,既然是共同k级祖先,那么直接找到a点的k级祖先,然后以这个点的子树,看看有多少深度为dep[now]+k的,个数-1就是和a是表亲关系的,对吧。其实转化一下,思路就很明显了。剩下的就是基本操作了。值得注意的是,如何求一个点的k级祖先呢,其实就是倍增往上跳就行了。对于这个k级,二进制拆分,如果这以为为1,那么就跳到这一位。acc[i][j]代表的就是i的2的j次方祖先。
代码:
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;
int T,n,m,k;
string va[N];
int siz[N],son[N],dep[N],acc[N][25],cnt = 22;
int sum[N];
int anw[N];
set<string > s[N];
vector<PII > v[N];
vector<int > e[N];
void get(int now,int p)
{
siz[now] = 1;
acc[now][0] = p;
dep[now] = dep[p]+1;
for(auto spot:e[now])
{
if(spot==p) continue;
get(spot,now);
siz[now] += siz[spot];
if(siz[spot]>siz[son[now]]) son[now] = spot;
}
}
void add(int now,int p,int value,int wson)
{
sum[dep[now]] += value;
for(auto spot:e[now])
{
if(spot==p||spot==wson) continue;
add(spot,now,value,wson);
}
}
void dfs(int now,int p,int del)
{
for(auto spot:e[now])
{
if(spot==p||spot==son[now]) continue;
dfs(spot,now,1);
}
if(son[now]) dfs(son[now],now,0);
add(now,p,1,son[now]);
for(auto t:v[now]) anw[t.se] = sum[dep[now]+t.fi]-1;
if(del) add(now,p,-1,0);
}
int query(int now,int x)
{
for(int i=cnt;i>=0;i--)
{
if(x>=1ll<<i)
{
x -= 1ll<<i;
now = acc[now][i];
}
}
return now;
}
signed main()
{
IOS;
cin>>n;
for(int i=1;i<=n;i++)
{
int a;
cin>>a;
e[a].pb(i);
acc[i][0] = a;
}
for(int i=1;i<=n;i++)
{
if(!acc[i][0]) get(i,0);
}
for(int i=1;i<=cnt;i++)
{
for(int j=1;j<=n;j++)
acc[j][i] = acc[acc[j][i-1]][i-1];
}
cin>>m;
for(int i=1;i<=m;i++)
{
int a,b;
cin>>a>>b;
int now = query(a,b);
v[now].pb({b,i});
}
for(int i=1;i<=n;i++)
{
if(!acc[i][0]) dfs(i,0,1);
}
for(int i=1;i<=m;i++) cout<<anw[i]<<" ";
return 0;
}
总结:
多多思考,理解本质。