Choosing Capital for Treeland CodeForces - 219D(树形DP)

传送门：QAQ

 

题意：有n个点，有n-1条有向边，使一个点可以到达其他所有点，可以翻转边的方向，问你最少翻转的边数，可以使其中一个点到达其他所有点（就是令一个点为根节点），然后让你升序输出每个点为根节点所需的翻转边数

 

思路：进行两次dfs，第一次我们用dp[v] += dp[u] + dir[u];（u为v的子节点，dir指两个点之间边的代价）处理出每个点为根结点时，需要翻转的边的数量

第二次我们依旧从之前深搜的方向，一点点去反转,dp[u] = dp[v] + (dir[u] ? -1 : 1);此时的dp[v]已经使以v为所有点的根节点的代价。

 

代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<stack>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
vector<int>gx[200100];
vector<int>f[200100];
int dp[200100];
int dir[200100];
void add(int u, int v) {
	gx[u].push_back(v);
	f[u].push_back(0);
	gx[v].push_back(u);
	f[v].push_back(1);
}
void dfs(int v, int pre) {
	dp[v] = 0;
	for (int i = 0; i < gx[v].size(); i++) {
		if (gx[v][i] == pre) continue;
		dir[gx[v][i]] = f[v][i];
		dfs(gx[v][i], v);
		dp[v] += dp[gx[v][i]] + dir[gx[v][i]];
	}
}
void solve(int v, int pre) {
	for (int i = 0; i < gx[v].size(); i++) {
		if (gx[v][i] == pre) continue;
		dp[gx[v][i]] = dp[v] + (dir[gx[v][i]] ? -1 : 1);
		solve(gx[v][i], v);
	}
}
int main(void) {
	int n;
	scanf("%d", &n);
	for (int i = 0; i < n - 1; i++) {
		int a, b;
		scanf("%d%d", &a, &b);
		add(a, b);
	}
	memset(dp, 0, sizeof(dp));
	dfs(1, -1);
	solve(1, -1);
	vector<int>q;
	int maxx = INF;
	for (int i = 1; i <= n; i++) {
		maxx = min(maxx, dp[i]);
	}
	for (int i = 1; i <= n; i++) {
		if (maxx == dp[i]) {
			q.push_back(i);
		}
	}
	printf("%d\n", maxx);
	for (int i = 0; i < q.size(); i++) {
		printf("%d ", q[i]);
	}
	printf("\n");
	return 0;
}