codeforces 219D Choosing Capital for Treeland (树形DP)

题目链接：https://vjudge.net/problem/CodeForces-219D

题意：给定一个树形有向图，需要求出从哪个点出发遍历整颗树需要改变的边的方向最少，输出最少的改变次数和符合条件的这些点。

很久没更新博客了，今天来水一发，首先建树完成后，从根节点出发dfs，用dp[i]记录从节点i出发遍历其子树所有节点所需要改变的最少次数，用vis[u][v]来保存边的方向，如果方向从u->v那么dp[u]+=dp[v]否则dp[u]+=dp[v]+1。然后再从根节点出发dfs，这次dfs需要更新dp数组，考虑由子节点v向父节点u遍历，如果边的方向从u->v，那么dp[v]=dp[u]+1否则dp[v]=dp[u]-1。本次更新完毕后dp数组就储存了从任意节点出发遍历整棵树的从小改变次数。然后找到答案输出即可。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<sstream>
#include<cmath>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<algorithm>
using namespace std;
const int MAXM=2*1e5;
const long long MOD=1000000007;
const int INF=1e9;
const double PI=acos(-1);

int n;
vector<int > e[MAXM+10];
int dp[MAXM+10];
vector<int > vis[MAXM+10];

void dfs1(int u,int pre) {
	int len=e[u].size();
	for (int i=0;i<len;i++)  {
		int v=e[u][i];
		if (v==pre) continue;
		dfs1(v,u);
		if (vis[u][i]) dp[u]+=dp[v];
		else dp[u]+=dp[v]+1;
		//cout<<u<<' '<<dp[u]<<endl;
	}
}
void dfs2(int u,int pre) {
	int len=e[u].size();
	for (int i=0;i<len;i++) {
		int v=e[u][i];
		if (v==pre) continue;
		if (vis[u][i]) dp[v]=dp[u]+1;
		else dp[v]=dp[u]-1;
		dfs2(v,u);
	}
}
int main()
{
	scanf("%d",&n);
	int u,v;
	for (int i=1;i<=n-1;i++) {
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
		e[v].push_back(u);
		vis[u].push_back(1);
		vis[v].push_back(0);
	}
	memset(dp,0,sizeof(dp));
	dfs1(1,-1);
	dfs2(1,-1);
	//for (int i=1;i<=n;i++)
		//cout<<dp[i]<<endl;
	int minm=INF;
	for (int i=1;i<=n;i++)
		if (dp[i]<minm) minm=dp[i];
	printf("%d\n",minm);
	int cnt=0;
	for (int i=1;i<=n;i++) {
		if (dp[i]==minm) {
			if (cnt!=0) printf(" ");
			printf("%d",i);
			cnt++;
		}
	}
	printf("\n");
	return 0;
}