青出于蓝 dfs序+树状数组

题目来源:蓝桥杯2018模拟

武当派一共有 n 人,门派内 n 人按照武功高低进行排名,武功最高的人排名第 1,次高的人排名第 2,... 武功最低的人排名第 n。现在我们用武功的排名来给每个人标号,除了祖师爷,每个人都有一个师父,每个人可能有多个徒弟。
我们知道,武当派人才辈出,连祖师爷的武功都只能排行到 p。也就是说徒弟的武功是可能超过师父的,所谓的青出于蓝胜于蓝。
请你帮忙计算每个人的所有子弟(包括徒弟的徒弟,徒弟的徒弟的徒弟....)中,有多少人的武功超过了他自己。
输入格式
输入第一行两个整数 n,p(1≤n≤100000,1≤p≤n)。
接下来 n-1 行,每行输入两个整数 u,v(1≤u,v≤n),表示 u 和 v 之间存在师徒关系。
输出格式
输出一行 n 个整数,第 i 个整数表示武功排行为 i 的人的子弟有多少人超过了他。
行末不要输出多余的空格。
样例输入
10 5
5 3
5 8
3 4
3 1
2 1
6 7
8 7
9 8
8 10
样例输出
0 0 2 0 4 0 1 2 0 0


首先用dfs对整个树型结构进行编号,以确定对应的孩子节点的个数情况,再用树状数组来维护,用了逆序对的思想,即用当比i的高的师傅和徒弟的总个数减去比i高的师傅的的已有个数,就是比i高的徒弟的个数,注意录入的时候按武功升序录入,保证了如果孩子有,就是比他高的。

#include
using namespace std;
const int maxn=100010;
int n,p,cnt,sj=0;//tm时间戳 
int head[2*maxn],l[maxn],r[maxn],c[maxn];
struct EG{
	int to,next;
}e[2*maxn];
void add(int a,int b){
	e[++cnt].to = b;
	e[cnt].next = head[a];
	head[a] = cnt;
}
void init(){
	memset(head,-1,sizeof head);
	scanf("%d %d",&n,&p);
	int a,b;
	for(int i=1;i

 

你可能感兴趣的:(例题,树形结构)