D. Decorate Apple Tree(记录每个点,叶子节点数目)

Problem - D - Codeforces

在Arkady的花园里有一棵苹果树。它可以表示为连接着枝干的节点集合，以便从任何一个节点到达任何其他节点时只有一种方法。节点从1到n进行编号，节点1称为根。

节点v的子树是指一组节点u，使得从u到根的路径必须经过v。请注意，v本身包含在其子树中。

叶子节点是这样一个节点，其子树恰好包含一个节点。

新年即将来临，所以Arkady想要装饰这棵树。他将在每个叶子节点上放置一个某种颜色的灯泡，然后计算出快乐节点的数量。快乐节点是这样一个节点t，其子树内的所有灯泡颜色都不相同。

Arkady感兴趣的问题是：对于1到n中的每个k，需要多少不同颜色的灯泡，才能使快乐节点的数量大于或等于k？

输入 第一行包含一个整数n（1≤n≤105）——树中节点的数量。

第二行包含n-1个整数p2、p3、…、pn（1≤pi<i），其中pi表示节点i和节点pi之间有一条分支。保证这些分支构成了一棵树。

输出 输出n个整数。第i个整数应该是使得快乐节点数量至少为i所需的最小颜色数。

 

Examples

input

Copy

3
1 1

output

Copy

1 1 2 

input

Copy

5
1 1 3 3

output

Copy

1 1 1 2 3 

注释 在第一个示例中，对于k=1和k=2，我们可以只使用一种颜色：节点2和3将是快乐的节点。对于k=3，你必须使用不同的颜色来使所有的节点都快乐。

在第二个示例中，对于k=4，例如，您可以在节点2和4上放置颜色为1的灯泡，并在节点5上放置颜色为2的灯泡。然后，快乐的节点是索引为2、3、4和5的节点。

题解:

注意我们只可以对叶子节点进行改变颜色,

首先很明显的一点是,假如有k个叶子节点,只需要一种颜色,就可以有k的快乐节点,因为每个叶子节点,都相当于一棵子树,

那么对于答案 > k的情况呢,由于要求最小改变颜色数,也就是说我们要找一个,最小的非叶子节点的点,这样需要改变的叶子节点数目最小,对于这个节点,看他有几个叶子节即可

依次类推

注意求每个点叶子节点的数目的代码,只和求每个点子树大小,有一点不同,那就是,遍历到一个新节点时,看看这个节点有没有子节点

#include <cstdio>
#include <cstring>
#include <algorithm>
#include<iostream>
#include<vector>
#include<set>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
#define int long long
typedef pair<int,int> PII;
int mod = 998244353;
int n;
int a[100050];
vector<int> p[100050];
void dfs(int x)
{
	if(!p[x].size())
	a[x] = 1;
	for(auto ne:p[x])
	{
		dfs(ne);
		a[x] += a[ne];
	}
}
void solve()
{
	int n;
	cin >> n;
	for(int i = 2;i <= n;i++)
	{
		int x;
		cin >> x;
		p[x].push_back(i);
	}
	dfs(1);
	sort(a + 1,a + 1 + n);
	for(int i = 1;i <= n;i++)
	cout << a[i] <<" ";
}
signed main()
{
//	ios::sync_with_stdio(0 );
//	cin.tie(0);cout.tie(0);
	int t = 1;
//	cin >> t;
	while(t--)
	{
		solve(); 
	}
}