CodeForces 238C World Eater Brothers 树形dp

题目链接：点击打开链接

题意：

给定n个点的有向树

下面n-1行给出正向的边。

问：

修改尽可能小的边的方向，可以任选2个起点使得这两个点bfs能遍历完所有点。

思路：

枚举一条边，把树分成两部分，每次计算这两部分的最小花费。

那么这样就变成一个子树上的dp。

对于一棵树所需要修改的最小边方向的方法：

首先容易求出：dp[i]表示以i为根的子树，用i作为起点的花费.

此时若以root为起点则代价就是cost[root] = dp[root], 若以树中任意一点u为起点

则代价就是cost[u] = dp[root] + (root->u方向的边数) - (u->root方向的边数)

那么为了得到最小的cost[i], 则就要使得(root->u方向的边数) - (u->root方向的边数)最大。

所以维护(root->u方向的边数) - (u->root方向的边数) 的最大值，最小的cost[i] = dp[root] - max;

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const int N = 1e5 + 10;
struct Edge{
	int from, to, dis, nex;
}edge[N<<1];
int head[N], edgenum;
void add(int u, int v, int d){
	Edge E = { u, v, d, head[u] };
	edge[edgenum] = E;
	head[u] = edgenum++;
}
int dp[N];
int n, mx;
void dfs(int u, int fa, int val){
	dp[u] = 0;
	for (int i = head[u]; ~i; i = edge[i].nex){
		int v = edge[i].to;
		if (v == fa)continue;
		dfs(v, u, val - edge[i].dis);
		dp[u] += dp[v] + (edge[i].dis!=1);
	}
	mx = max(mx, val);
}
int main(){
	while (cin >> n){
		if (n == 1){ puts("0"); continue; }
		memset(head, -1, sizeof head);
		edgenum = 0;
		for (int i = 1, u, v; i < n; i++){
			scanf("%d %d", &u, &v);
			add(u, v, 1);
			add(v, u, -1);
		}
		int ans = 1 << 30;
		for (int i = 0, u, v; i < edgenum; i += 2){
			u = edge[i].from; v = edge[i].to;
			mx = -(1<<30);
			dfs(u, v, 0);
			int tmp = dp[u] - mx;

			mx = -(1<<30);
			dfs(v, u, 0);
			tmp += dp[v] - mx;
			ans = min(ans, tmp);
		}
		printf("%d\n", ans);
	}
	return 0;
}