cf 135div2D Choosing Capital for Treeland 【树形dp】

链接：http://codeforces.com/problemset/problem/219/D

题意：n个点，m个有向边，现在要你求出每个点作为根使得图为一棵树需要改变多少边的方向，输出最少改变多少次，和这些点。

分析：建图时发现，一个点为根时他的出度边不用改变，所以将所有的入边的权值为1，然后用dfs统计每个点的子树有多少这样的边。

我们还可以发现，一个点（u）和他的父亲（fa）的关系，就是u成为整个树的root时，答案的改变只和（u，fa）这条边有关，在 第二次dfs是可以更新。

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define Mn 200005
#define Mm 400010
#define mod 1000000007
#define CLR(a,b) memset((a),(b),sizeof((a)))
#define CLRS(a,b,Size) memset((a),(b),sizeof((a[0]))*(Size+1))
#define CPY(a,b) memcpy ((a), (b), sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)|1
using namespace std;
typedef long long ll;
struct edge {
    int v,w,next;
}e[Mm];
int tot=0,head[Mn];
void addedge(int u,int v,int w) {
    e[tot].v=v;
    e[tot].w=w;
    e[tot].next=head[u];
    head[u]=tot++;
}
int dp[Mn];
void dfs(int u,int fa) {
    for(int i=head[u];~i;i=e[i].next) {
        int v=e[i].v;
        int w=e[i].w;
        if(v==fa) continue;
        dfs(v,u);
       // cout<<u<<" "<<v<<" "<<dp[v]<<" "<<w<<endl;
        dp[u]+=dp[v]+w;
    }
}
int minn;
void dfs2(int u,int fa) {
    minn=min(dp[u],minn);
    for(int i=head[u];~i;i=e[i].next) {
        int v=e[i].v;
        int w=e[i].w;
        if(v==fa) continue;
        if(w) dp[v]=dp[u]-1;
        else dp[v]=dp[u]+1;

        dfs2(v,u);
    }
}
void init() {
    CLR(head,-1);
    tot=0;
}
int main() {
    int n;
    init();
    scanf("%d",&n);
    minn=n;
    for(int i=1;i<n;i++) {
        int u,v;
        scanf("%d%d",&u,&v);
        addedge(u,v,0);
        addedge(v,u,1);
    }
    dfs(1,0);
    dfs2(1,0);
    printf("%d\n",minn);
    for(int i=1;i<=n;i++) {
        if(dp[i]==minn)
        printf("%d ",i);
    }
    return 0;
}