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;
}