Codeforces Round #135 (Div. 2)D. Choosing Capital for Treeland

题意：n个点，n-1个边，有向的，问当以某个点为起点，到其他点的话要改变多少个边的方向

思路：题目给的边价值为0，反向的价值为1，f[i]表示该点到他的子节点需要改变的边，g[i]为到他上方需要改变的，跑2遍图

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;

struct node{
    int to,next,val;
}e[N*2];
int head[N],tot;
int f[N],g[N];

void init(){
    memset(head,-1,sizeof(head));
    tot=0;
}

void add(int u,int v,int w){
    e[tot].to=v;e[tot].next=head[u];e[tot].val=w;
    head[u]=tot++;
}

void dfs1(int u,int fa){
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to,w=e[i].val;
        if(v==fa) continue;
       // cout<<u<<" "<<v<<" "<<w<<endl;
        dfs1(v,u);
        f[u]+=f[v]+w;
    }
}
int Min=1e9+7;
void dfs2(int u,int fa){
    Min=min(Min,f[u]+g[u]);
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to,w=e[i].val;
        if(v==fa) continue;
        g[v]=g[u]+f[u]-f[v]+(w==0?1:-1);
        dfs2(v,u);
    }
}

int main(){
    int n;
    int x,y;
    init();
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        scanf("%d%d",&x,&y);
        add(x,y,0);
        add(y,x,1);
    }
    dfs1(1,0);
    dfs2(1,0);
    cout<<Min<<endl;
    for(int i=1;i<=n;i++){
        if(f[i]+g[i]==Min)
            printf("%d ",i);
    }
    printf("
");
}