CodeForces 219D-H - Choosing Capital for Treeland-树DP

http://codeforces.com/problemset/problem/219/D

题意：

给出一树的节点n和有向边n-1条，

对每个点有个信息：该点到所有其他点，如果遇到一条反向边，计数器+1，

求出，最后这个计数器最小的点，如果有多个，增序输出

题目本身是无根树，为了方便我们直接看成一个以1为根的有根树即可：

显然是一个树DP，

我们把正向边权值val为0，反向边权值为1

第一次dfs，自底向上求出每个节点的到所有叶子节点所遇到的反向边数之和  dn[x]

然后对每个节点，要求题目所要求的信息，我们还缺【该节点到根节点的反向边数量，和 到兄弟节点的反向边数量】

到根结点的好求，我们只需要再来个dfs2，这次自上向下，每到达一个节点，给up[x]加上从根到当前节点的反向边数量。

然后对于到兄弟节点的反向边的话需要利用父亲节点的信息，就等于 dn[x]-dn[v]-(!val)  【val是边权】 

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;

const double pi=acos(-1.0);
double eps=0.000001; 
int min(int a,int b)
{return a<b?a:b;}
struct node
{
	int x,v,id;
	node(int a=0,int b=0,int c=0)
	{x=a,v=b;id=c;}
};

vector<node> mp[200005];
int up[200005];
int dn[200005];
int vis[200005];
int ans[200005];
int ok=0; 
void dfs2(int x )
{ 
	vis[x]=1;
	int i;
	for (i=0;i<mp[x].size();i++)
	{
		int v=mp[x][i].x;
		int val=mp[x][i].v; 
		val=!val;
		if (vis[v])continue;
		up[v]=val+dn[x]-dn[v]-(!val)+up[x];
		dfs2(v);   
	}
}
int dfs1(int x )
{ 
	int i;vis[x]=1;
	for (i=0;i<mp[x].size();i++)
	{
		int v=mp[x][i].x;
		int val=mp[x][i].v;
		if (vis[v])continue;
		dn[x]+=val;
		dn[x]+=dfs1(v);
	} 
	return dn[x]; 
} 
int main()
{
	int n;
	cin>>n; 
	int i,j;
	int x,y;
	for (i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		mp[x].push_back(node(y,0,i));
		mp[y].push_back(node(x,1,i));
	}
	
	dfs1(1);
	memset(vis,0,sizeof(vis));
	dfs2(1);
	int minn=1e9,maxi=-1;
for (i=1;i<=n;i++)  
	minn=min(minn,up[i]+dn[i]); 

 
for (i=1;i<=n;i++) 
{
	if (minn==up[i]+dn[i])
	ans[++ok]=i;
} 
  printf("%d\n",minn);
	for (i=1;i<=ok;i++)
		printf("%d ",ans[i]);
	printf("\n"); 

	
	return 0;
}