Codeforces Round #135 (Div. 2) D. Choosing Capital for Treeland(从每个点出发最小需要修改几条边的方向可以到达其它所有点,输出最小改变几)

题意:
有n个点的树(n<=2e5),每条边定向,问从每个点出发最小需要修改几条边的方向可以到达其它所有点,输出最小改变几条边,
以及对应点

---

思路:
先不给树定向,向下dfs,向上dfs判断

---

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+100;
struct Edge{
    int to,next,w;
}e[N*2];
int tot,head[N],n,down[N],up[N];

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

void addedge(int from,int to,int dir){  //dir为0表示从from到to,dir为1表示从to到from
    e[tot]=(Edge){to,head[from],dir};
    head[from]=tot++;
}

void dfs1(int u,int pre){
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v==pre)  continue;
        dfs1(v,u);
        down[u]+=down[v]+e[i].w;
    }
}

void dfs2(int u,int pre){
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v==pre)  continue;
        up[v]=up[u]+(down[u]-down[v]-e[i].w)+(1-e[i].w);
        dfs2(v,u);
    }
}

int main(){
    int n,u,v;
    scanf("%d",&n);
    init();
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        addedge(u,v,0);
        addedge(v,u,1);
    }
    dfs1(1,0);
    dfs2(1,0);
    int mx=n;
    for(int i=1;i<=n;i++)
        mx=min(down[i]+up[i],mx);
    printf("%d\n",mx);
    for(int i=1;i<=n;i++)
        if(down[i]+up[i]==mx)
            printf("%d ",i);
    printf("\n");
    return 0;
}