武当派一共有 n 人,门派内 n 人按照武功高低进行排名,武功最高的人排名第 1,次高的人排名第 2,... 武功最低的人排名
第 n。现在我们用武功的排名来给每个人标号,除了祖师爷,每个人都有一个师父,每个人可能有多个徒弟。
我们知道,武当派人才辈出,连祖师爷的武功都只能排行到 pp。也就是说徒弟的武功是可能超过师父的,所谓的青出于蓝胜于蓝。
请你帮忙计算每个人的所有子弟(包括徒弟的徒弟,徒弟的徒弟的徒弟....)中,有多少人的武功超过了他自己。
输入格式
输入第一行两个整数 n, p(1 ≤ n ≤ 100000, 1 ≤ p ≤ n) n , p (1≤n≤100000,1≤p≤n)。
接下来 n-1 行,每行输入两个整数 u, v(1 ≤ u, v ≤ n)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
题意:对以p为根的树,找有多少个儿子节点比他小。
思路:总体思路是利用dfs序+树状数组(或者线段树也是行得通的)这里只讨论树状数组的做法,线段树的思路类似,都是用
来维护一个单点操作+区间求和的问题。 时间复杂度为 O(nlogn)
首先,直接在树上操作看起来不大方便,所以可以先已p为根节点求一个dfs序,那样我们就得到一个序列。
比如,根据上图我们可以得到一个dfs序:5,3,4,1,2,8,7,6,9,10
很明显,对于每一个节点,它的儿子节点在其dfs序中肯定是位于本节点之后,且是连续的。
比如节点3,节点3有3个儿子节点,根据dfs序,子节点就是4,1,2。
所以我们可以先求出dfs序,记录每一个节点的子节点个数,那这道题就可以转变为:
给你一个序列,求以每一个节点开始的n个数(当前节点的子节点数),有多少个数小于它。
对于上面这个问题,很明显可以想到求逆序数那一题,只要利用树状数组求解一下就好了,具体方法就是:将序列从小到大以权
值1加入到树状数组中,插入后只需要查询当前要求区间( [i,i+k-1] ,k为节点i的子节点数) 的和,(因为是从小到大加入的,
所以所得求和就是区间中必当前值小的数的个数)。
还不大清楚的可以先看一下求逆序数的思路,再来看这道题。
代码如下:
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
int dfn[100050],sum[100050],vis[100050],cnt = 0;
int n,b[100050],c[100050],res[100050];
vector<int> E[100050];
void addedge(int u,int v)
{
E[u].push_back(v);
E[v].push_back(u);
}
int dfs(int root) //求dfs序
{
dfn[++cnt] = root;
int nums = E[root].size();
int tot = 0;
for(int i=0;i<nums;i++)
{
int v = E[root][i];
if(!vis[v])
{
vis[v] = 1;
tot += dfs(v);
}
}
return sum[root] = tot + 1;
}
int lowbit(int x)
{
return x&(-x);
}
void add(int pos,int x)
{
while(pos<=n)
{
c[pos]+=x;
pos += lowbit(pos);
}
}
int query(int pos) //树状数组求和(1-pos) 注意树状数组必须从1开始
{
int tot = 0;
while(pos>0)
{
tot += c[pos];
pos -= lowbit(pos);
}
return tot;
}
void work(int root)
{
vis[root] = 1;
sum[root] = dfs(root);
for(int i=1;i<=n;i++) b[dfn[i]] = i;
for(int i=1;i<=n;i++)
{
res[i] = query(b[i]+sum[i]-1) - query(b[i]);
add(b[i],1);
}
for(int i=1;i<=n;i++)
{
printf("%d",res[i]);
if(i!=n) printf(" ");
}
printf("\n");
}
int main()
{
#ifdef local
freopen("in.txt","r",stdin);
#endif // local
int p; scanf("%d%d",&n,&p);
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
addedge(u,v);
}
work(p);
}