CF 1056D Decorate Apple Tree

原创于 2022-01-13 17:17:47 发布 · 201 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树结构

codeforces 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种使用树形结构进行动态规划的问题解决方法。通过挂灯泡的例子详细讲解了如何计算最少所需颜色数量以确保特定数量的节点成为“happy”节点。文章提供了完整的代码实现，包括深度优先搜索算法的应用。

在这里插入图片描述

题目大意：
有一颗树，可以往节点上挂灯泡（有很多种颜色），定义一个happy节点：这颗节点子树的所有叶子节点所挂的灯泡颜色不同
现在问你有1到n个happy节点，分别需要最小多少种颜色来装饰灯泡

1.一个结点可以代表一颗子树，n个结点可以代表n个子树
2.一个happy结点的cost为该子树的叶子结点数
不难得到，k个happy结点，即用前k小的cost的结点
可以dfs,得到所有结点的叶子结点，然后排序，从小到大输出即可

const int N = 1e6+10;

vector<int> e[N];
int f[N];

void dfs(int u){
    if(e[u].size()==0){
        f[u] = 1;
        return;
    }
    for(auto v:e[u]){
        dfs(v);
        f[u] += f[v];
    }
}

signed main(){
    int n;
    cin >> n;
    
    For(i,2,n){
        int t;
        scanf("%d", &t);
        e[t].push_back(i);
    }

    dfs(1);
    sort(f + 1, f + 1 + n);

    For(i,1,n)printf("%d ",f[i]);
}