青出于蓝 dfs序+树状数组

该博客围绕蓝桥杯2018模拟题展开,题目是计算武当派每个人的子弟中武功超过自己的人数。给出输入输出格式及样例,解题思路是用dfs对树形结构编号,用树状数组维护,运用逆序对思想,录入时按武功升序录入。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目来源:蓝桥杯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<bits/stdc++.h>
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<n;i++){
		scanf("%d %d",&a,&b);
		add(a,b);
		add(b,a);
	}
}
void dfs(int s,int fa){
	l[s]=++sj;
	for(int i=head[s];~i;i=e[i].next){
		if(e[i].to!=fa){
			dfs(e[i].to,s);
		}
	}
	r[s]=sj;
}
int lowbit(int x){
	return x&(-x);
}
int sum(int x){
	int ans=0;
	for(int i=x;i;i-=lowbit(i)){
		ans+=c[i];
	}
	return ans;
}
void updata(int pos,int k){
	for(int i=pos;i<=n;i+=lowbit(i))
	c[i]+=k;
}
void solve(){
	for(int i=1;i<=n;i++){
		printf("%d%c",sum(r[i])-sum(l[i]),i==n?'\n':' ');//假设r[i]=5,l[i]=2,即从0-5的已有个数(比i的高的师傅和徒弟的总个数)减去0-2(比i高的师傅的)的已有个数,
		                                                 //就是比i高的徒弟的个数
		updata(l[i],1);//只有武功比i高的才会被先录入, 
		               
	}
}
int main(){
	init();
	dfs(p,-1);
	solve();
	return 0;
} 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值