codeforces 1336A Linova and Kingdom

题意

链接: link.
一个王国
有编号1到n共n个城市，其中1是首都
有n-1条公路，树形结构的一个图
需要建立m个工业城市，工业城市的贡献为首都到工业城市之间存在的非工业城市数量，问总的贡献最大是多少？

思路

1、整体思路，选择叶子更优
2、假设1~n形成一条链，如果m=1，选择叶子结点最优
m=2，我们应该选择最后两个节点
当一棵树有2条链，其中一条链很长，其中一条只有节点长度为1的，我们还是选择长链末端两个。所以补充一点，我们并不是选择所有叶子节点最优，我们应该选择深度最大的节点作为工业城市
3、假设有深度相同的两个节点，其中a节点有两子节点，b只有一个叶子节点，那么我们应该优先选择a
两条深度相同的ab节点，a有2叶子节点，b有1叶子节点，此时ab叶子节点均已经被占领作为工业城市，此时要在ab中选择一节点作为新工业城市的话，应该选择b,因为选择ab节点深度一样，当需要选择此节点作为工业城市的时候没说明他们的叶子节点已经是工业城市，把a作为工业城市，则a城市的叶子节点减少2贡献，b减少1贡献，也就是当父节点作为工业城市的时候，旗下有多少子节点就会减少多少贡献，应该尽量选择子节点少的节点
4当一棵树固定之后，每一个节点的贡献就已经固定了
总结，该节点贡献 = 该节点深度 - 其子树节点个数
我们贪心选择前m个贡献最大的节点即可.

AC代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2E5 + 10;
vector<int> edge[N];//邻接表
int qaq[N], sz[N];//每一个节点的贡献//
void dfs(int x = 1, int fa = 0, int depth = 0)
{
    qaq[x] = depth;
    sz[x] = 1;

    for (auto &to : edge[x])
    {
        if (to == fa)
            continue;
        dfs(to, x, depth + 1);
        sz[x] += sz[to];
    }

    qaq[x] -= sz[x] - 1;
}
int main()
{
    int n, m;
    cin >> n >> m;
    for (int i = 1; i < n; i++)
    {
        int a, b;
        scanf("%d %d", &a, &b);
        edge[a].push_back(b);
        edge[b].push_back(a);
    }

    dfs();

    sort(qaq + 1, qaq + 1 + n, greater<>());
    ll res = 0;
    for (int i = 1; i <= m; i++)
    {
        res += qaq[i];
    }
    cout << res << endl;
    return 0;
}