青出于蓝胜于蓝-(树状数组+DFS序)

武当派一共有 nn 人,门派内 nn 人按照武功高低进行排名,武功最高的人排名第 11,次高的人排名第 22,... 武功最低的人排名第 nn。现在我们用武功的排名来给每个人标号,除了祖师爷,每个人都有一个师父,每个人可能有多个徒弟。

我们知道,武当派人才辈出,连祖师爷的武功都只能排行到 pp。也就是说徒弟的武功是可能超过师父的,所谓的青出于蓝胜于蓝。

请你帮忙计算每个人的所有子弟(包括徒弟的徒弟,徒弟的徒弟的徒弟....)中,有多少人的武功超过了他自己。

输入格式

输入第一行两个整数 n, p(1 \le n \le 100000, 1 \le p \le n)n,p(1n100000,1pn)

接下来 n-1n1 行,每行输入两个整数 u, v(1 \le u, v \le n)u,v(1u,vn),表示 uu 和 vv 之间存在师徒关系。

输出格式

输出一行 nn 个整数,第 ii 个整数表示武功排行为ii 的人的子弟有多少人超过了他。

行末不要输出多余的空格


题解:首先利用dfs将树上问题化为一维问题,然后用树状数组按照逆序对的求法更新加求答案即可。(很水的。。)

#include<vector>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define maxn 100005
vector<int>q[maxn];
int n,p,a[maxn],b[maxn],st,ed,c[maxn],ans[maxn];
void dfs(int u,int p)
{
	a[u]=++st;
	for(int i=0;i<q[u].size();i++)
	{
		int v=q[u][i];
		if(v==p)
			continue;
		dfs(v,u);
	}
	b[u]=st;
}
int find(int x)
{
	int res=0;
	while(x)
		res+=c[x],x-=x&-x;
	return res;
}
void update(int x)
{
	while(x<=n)
		c[x]++,x+=x&-x;
}
int main(void)
{
	int x,y;
	scanf("%d%d",&n,&p);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		q[x].push_back(y);
		q[y].push_back(x);
	}
	dfs(p,0);
	for(int i=n;i>0;i--)
	{
		ans[i]=b[i]-a[i]-(find(b[i])-find(a[i]));
		update(a[i]);
	}
	for(int i=1;i<n;i++)
		printf("%d ",ans[i]);
	printf("%d\n",ans[n]);
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值