[codeforces219D]Choosing Capital for Treeland树形dp

题意：给出一棵树，带有向边，找出某个点到达所有点需要反转的最少的边。

解题关键：和求树的直径的思路差不多，将求（父树-子树）的最大值改为求特定值。依然是两次dfs，套路解法。

对树形dp的理解：树形dp其实就是将树进行暴力搜索，只是需要理解状态的概念。那些状态已经完成，需要从底还是从顶开始搜索。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn=1e6+3;
const int inf=0x3f3f3f3f;
int head[maxn],tot,n,m,sum;
int dp[maxn][3],cnt[maxn];
struct edge{
    int to;
    int nxt;
    int w;
}e[maxn<<2];
void add_edge(int u,int v,int w){
    e[tot].to=v;
    e[tot].w=w;
    e[tot].nxt=head[u];
    head[u]=tot++;
}

void dfs1(int u,int fa){
    for(int i=head[u];i!=-1;i=e[i].nxt){
        int v=e[i].to;
        if(v==fa) continue;
        dfs1(v,u);
        dp[u][0]+=dp[v][0]+e[i].w;
    }
}

void dfs2(int u,int fa){
    for(int i=head[u];i!=-1;i=e[i].nxt){
        int v=e[i].to;
        if(v==fa) continue;
        dp[v][1]=dp[u][0]-dp[v][0]+dp[u][1];
        if(e[i].w==0) dp[v][1]+=1;
        else dp[v][1]-=1;
        dfs2(v,u);
    }
}

inline int read(){
    char k=0;char ls;ls=getchar();for(;ls<'0'||ls>'9';k=ls,ls=getchar());
    int x=0;for(;ls>='0'&&ls<='9';ls=getchar())x=(x<<3)+(x<<1)+ls-'0';
    if(k=='-')x=0-x;return x;
}

int main(){
    int k=0;
    while(scanf("%d",&n)!=EOF){
        memset(head,-1,sizeof head);
        tot=0;
        int a,b;
        /*for(int i=1;i<n;i++){
            cnt[i]=read();
            sum+=1ll*cnt[i];
        }*/
        for(int i=0;i<n-1;i++){
            a=read();
            b=read();
            add_edge(a,b,0);
            add_edge(b,a,1);
        }
        dfs1(1,-1);
        dfs2(1,-1);
        for(int i=1;i<=n;i++){
            dp[i][2]=dp[i][0]+dp[i][1];
        }
        int min1=inf;
        for(int i=1;i<=n;i++){
            min1=min(dp[i][2],min1);
        }
        cout<<min1<<"
";
        for(int i=1;i<=n;i++){
        //    cout<<dp[i][2]<<" ";
            if(dp[i][2]==min1) cout<<i<<" ";
        }
    }
    return 0;
    return 0;
}