CF219D Choosing Capital for Treeland(换根dp模板 一个点可任意到达其他节点需要翻转的边数)

题目
dp[x]：x想到达所有节点需要翻转的边数。
先求出root=1想到所有节点需要翻转的边数。自底向上dfs1。
然后自顶详细dfs2。
if(fa->y) dp[y]=dp[fa]+1;
if(y->fa) dp[y]=dp[fa]-1;(因为本身fa想到达所有的话就必须翻转过来与y这个儿子的 但儿子想到达所有正好这个边可以到达fa emm不多说了 很好理解的对吧)

#include<bits/stdc++.h>
#define en '\n'
#define m(a,b) memset(a,b,sizeof a)
using namespace std;
typedef long long ll;
const int N=2e5+5;
struct Edge{int to,nex,flag;}edge[N<<1];
int head[N],tot;
void add(int from,int to){
    edge[++tot]=(Edge){to,head[from],1};head[from]=tot;
    edge[++tot]=(Edge){from,head[to],0};head[to]=tot;
}
int dp[N];
int dfs1(int x,int fa){//自底向上   dfs1只为得到dp[1]
    int res=0;
    for(int i=head[x];i;i=edge[i].nex){
        int y=edge[i].to;
        if(y==fa) continue;
        res+=dfs1(y,x)+(edge[i].flag?0:1);
    }
    return res;
}
void dfs2(int x,int fa){//自顶向下  得到所有的答案
    for(int i=head[x];i;i=edge[i].nex){
        int y=edge[i].to;
        if(y==fa) continue;
        dp[y]=dp[x]+(edge[i].flag?1:-1);
        dfs2(y,x);
    }
}
int main(){
    int n;
    while(~scanf("%d",&n)){
        m(head,0),tot=0;
        for(int i=1;i<=n-1;++i){
            int from,to;scanf("%d%d",&from,&to);
            add(from,to);
        }
        dp[1]=dfs1(1,1);
        dfs2(1,1);
        int ans=n-1;
        for(int i=1;i<=n;++i) ans=min(ans,dp[i]);
        printf("%d\n",ans);
        for(int i=1;i<=n;++i) if(dp[i]==ans) printf("%d ",i);
        puts("");
    }
}