题意:给一个树形图(有向树),n个节点,n-1条有向边,要求选一个节点作为根,使需要改变方向的边的数目最少。并输出所有可能作为根的点。
设dp[i]表示以i为根,需要改变多少个边方向
题解:首先dfs从下向上dp[i],然后在dfs一次从上更新dp[i]
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 5;
int head[maxn], tot, dp[maxn];
const int inf = 1e9;
int mm = inf;
struct Edge {
int u, v, opt, next;
}edge[maxn];
void init() {
memset(head, -1, sizeof(head)); tot = 1;
}
void add(int u, int v, int opt) {
edge[++tot].u = u; edge[tot].v = v;
edge[tot].opt = opt; edge[tot].next = head[u];
head[u] = tot;
}
void dfs1(int u, int fa) {
for (int i = head[u]; i != -1; i = edge[i].next) {
int to = edge[i].v;
if (to == fa) continue;
dfs1(to, u);
dp[u] += dp[to] + (edge[i].opt == 0);
}
}
void dfs2(int u, int fa) {
mm = min(mm, dp[u]);
for (int i = head[u]; i != -1; i = edge[i].next) {
int to = edge[i].v;
if (to == fa) continue;
if (edge[i].opt == 1) {
dp[to] = dp[u] + 1;
} else {
dp[to] = dp[u] - 1;
}
dfs2(to, u);
}
}
int main() {
int n, u, v;
scanf("%d", &n);
init();
for (int i = 1; i <= n - 1; i++) {
scanf("%d%d", &u, &v);
add(u, v, 1); add(v, u, 0);
}
dfs1(1, 0);
// for (int i = 1; i <= n; i++) cout << dp[i] << " "; cout << endl;
dfs2(1, 0);
// for (int i = 1; i <= n; i++) cout << dp[i] << " "; cout << endl;
printf("%d\n", mm);
for (int i = 1; i <= n; i++) {
if (dp[i] == mm) {
printf("%d ", i);
}
}
return 0;
} 
京公网安备 11010502036488号