CodeForces 219D Choosing Capital for Treeland (树形dp)

题意:在一颗树中,边都是单向的,找到一个节点,使得它如果能到达任何一个结点的话,反转的边最少。

题解:树形dp
进行两次深搜。
第一次深搜,用dp[u][0]表示从节点u出发到达任一点所要反转的边数。
第二次深搜,已知dp[root][0]表示的就是root点的答案,继续深搜更新子节点,对于fa和son,fa始终是更新完成的节点(即答案),先让fa的答案赋值到son,那么仅需要考虑fa和son之间的一条边,+1-1即可。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
const int maxn = 2e5 + 5;
using namespace std;
int dp[maxn][2], k, head[maxn], n, u, v;
struct node {
	int v, next, w;
}edge[maxn * 2];
void add(int u, int v, int w) {
	edge[++k].next = head[u];
	edge[k].v = v;
	edge[k].w = w;
	head[u] = k;
}
void dfs1(int u, int fa) {
	for (int i = head[u]; i; i = edge[i].next) {
		int v = edge[i].v;
		if (v != fa) {
			dfs1(v, u);
			dp[u][0] += dp[v][0] + edge[i].w;
		}
	}
}
void dfs2(int u, int w, int fa){
	if (u == fa) dp[u][1] = dp[u][0];
	else {
		if (w) dp[u][1] = dp[fa][1] - 1;
		else dp[u][1] = dp[fa][1] + 1;
	}
	for (int i = head[u]; i; i = edge[i].next) {
		int v = edge[i].v;
		if (v != fa) dfs2(v, edge[i].w, u);
	}
}
int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n - 1; i++) {
		scanf("%d%d", &u, &v);
		add(u, v, 0);
		add(v, u, 1);
	}
	dfs1(1, -1);
	dfs2(1, -1, 1);
	int ans = maxn;
	for (int i = 1; i <= n; i++) {
		ans = min(ans, dp[i][1]);
	}
	printf("%d\n", ans);
	for (int i = 1; i <= n; i++) {
		if (dp[i][1] == ans) printf("%d ", i);
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值