这道题综合运用了搜索和树状数组
原理如下
先DFS一下,搜索一整棵树,将树的每一个节点标记上时序(即DFS搜索时候的顺序)
记录两个,一个是往下搜的时序,一个是往回回溯的时序,这样这两个标记可以表示这个节点的全部子节点有多少个,有哪些
节点(比如标记为 1 3 ,那么子节点就有2 3)
建立一个空的树状数组。
然后从武当派的第一名开始,将人一个一个的添加到树状数组中
每次添加一个人,那么就把这个人的师傅的树状数组的C[i]加上1(包括师傅的师傅,师傅的师傅的师傅...直到祖师爷)
查询这个人有多少徒弟超过了自己,那么就用回溯的标记减去下搜的标记
#include<bits/stdc++.h>
using namespace std;
const int mmax = 100010;
struct node//存图
{
int en,next;
} E[2 * mmax];
int p[mmax];
int num;
void init()
{
memset(p,-1,sizeof p);
num = 0;
}
void add(int st,int en)
{
E[num].en = en;
E[num].next = p[st];
p[st] = num++;//
}
int l[mmax],r[mmax];
int times;
void dfs(int u,int fa)
{
l[u] = ++times;//遍历时计算遍历次序
for(int i = p[u]; i+1; i = E[i].next)
{
int v = E[i].en;
if(v != fa)
{
dfs(v,u);
}
}
r[u] = times;
}
int n,f;
int C[mmax];
void up(int x,int v)//树状数组的添加,往l[x]的后面添加1
{
for(int i = x; i <= n; i += (i & -i))
{
C[i] += v;
}
}
int sum(int x)//树状数组计算前和
{
int res = 0;
for(int i = x; i > 0; i -= (i & -i))
{
res += C[i];
}
return res;
}
int main()
{
init();
scanf("%d%d",&n,&f);
for(int i = 0; i < n - 1; ++i)
{
int u,v;
cin >> u >> v;//输入
add(u,v);//双向存图
add(v,u);
}
times = 0;//times表示存图时的时序
dfs(f,-1);
for(int i = 1; i <= n; ++i)
{
cout << sum(r[i]) - sum(l[i]) << " \n"[i == n];
up(l[i],1);
}
return 0;
}