Codeforces 219D Choosing Capital for Treeland

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

题目大意：

给出一棵树，但是它的边是有向边，选择一个城市，问最少调整多少条边的方向能使一个选中城市可以到达所有的点，输出最小的调整的边数，和对应的点。

思路:先预处理一个点为根的代价，然后去dfs移动，总复杂度是O(n)

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<iostream>
int tot,go[400005],next[400005],first[200005],id[400005];
int son[200005],f[200005],v[200005],n;
int read(){
    int t=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while ('0'<=ch&&ch<='9'){t=t*10+ch-'0';ch=getchar();}
    return t*f;
}
void insert(int x,int y,int Id){
    tot++;
    go[tot]=y;
    next[tot]=first[x];
    first[x]=tot;
    id[tot]=Id;
}
void add(int x,int y){
    insert(x,y,1);insert(y,x,-1);
}
void dfs(int x,int fa){
    son[x]=0;
    for (int i=first[x];i;i=next[i]){
        int pur=go[i];
        if (pur==fa) continue;
        if (id[i]==-1) v[pur]=1;
        dfs(pur,x);
        son[x]+=son[pur]+v[pur];
    }
}
void dp(int x,int fa){
    for (int i=first[x];i;i=next[i]){
        int pur=go[i];
        if (pur==fa) continue;
        if (id[i]==1) f[pur]=f[x]+1;
        else f[pur]=f[x]-1;
        dp(pur,x);
    }
}
int main(){
    n=read();
    for (int i=1;i<n;i++){
        int x=read(),y=read();
        add(x,y);
    }
    dfs(1,0);
    f[1]=son[1];
    dp(1,0);
    int ans=n;
    for (int i=1;i<=n;i++) ans=std::min(ans,f[i]);
    printf("%d
",ans);
    for (int i=1;i<=n;i++) if (ans==f[i])
     printf("%d ",i);
    return 0;
}