P2018 消息传递

暴力贪心即可.
考虑从下往上计算. 对于一个根节点的所有子节点, 明显我们应该优先告诉耗时长的子节点.
于是贪心. 假设已经知道了所有子节点的最少花费时间. 我们将其从小到大排序, 那么最后一个应该最先告诉.这可以通过交换法证明.
当前节点的最小花费也就可以构造出来了.

#include <vector>
#include <cstdio>
#include <cstring>
#include <cassert>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN = 1e3 + 10;
inline int read(){
    char ch = getchar(); int x = 0;
    while(!isdigit(ch)) ch = getchar();
    while(isdigit(ch)) x = x * 10 + ch - '0', ch = getchar();
    return x;
}

int N; vector<int> g[MAXN];

int dp(int u, int fa) {
    int tmp = 0; vector<int> nodes;
    for(int i = 0; i < (int) g[u].size(); i++) {
        int &v = g[u][i]; if(v == fa) continue;

        nodes.push_back(dp(v, u));
    }
    sort(nodes.begin(), nodes.end());
    for(int i = 0; i < (int) nodes.size(); i++)
        tmp = max(tmp, nodes[i] + ((int)nodes.size() - i));
    return tmp;
}

int main(){
    cin>>N;
    for(int i = 2; i <= N; i++) {
        int v = read();
        g[i].push_back(v), g[v].push_back(i);
    }

    int ans = (1 << 30); vector<int> res;
    for(int i = 1; i <= N; i++) {
        int tmp = dp(i, i);
        if(tmp < ans) {
            ans = tmp; res.clear(); res.push_back(i);
        }
        else if(tmp == ans) res.push_back(i);
    }
    sort(res.begin(), res.end());
    printf("%d\n", ans + 1);
    for(int i = 0; i < (int) res.size(); i++)
        printf("%d ", res[i]);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/wsmrxc/p/9858056.html
今日推荐