Codeforces 219D

题目链接：https://vjudge.net/problem/CodeForces-219D

首先，类似于网上大部分题解所说的，设某一条边正向的weight=0,反向的weight=1，因为反向意味着我们（按DFS方向）走到这条边的时候，就得把这条边翻转，就相当于记一次（w=1）。

然后同样的，

dp[i][0]：表示以i为root的子树中，我们从i出发，要翻转（inverse）多少条边。

dp[i][1]：表示从i往其父亲方向走，要翻转（inverse）多少条边。

那么，最后我们要的，当一个点i为capital时，就要翻转dp[i][0]+dp[i][1]条边。

类似于HDU2196（http://www.cnblogs.com/dilthey/p/7186556.html），我们也要分两次做DFS，第一次做出所有的dp[i][0]，第二次做出所有的dp[i][1]；

状态转移方程：

对于某个节点u，有子节点（直接相连的）v1~vk，则

dp[u][0] = ( edge[u][v1].w + dp[v1][0] ) + …… + ( edge[u][vk].w + dp[vk][0] )

对于某个节点v，有父亲节点u，则

dp[v][1] = dp[u][1] + ( dp[u][0] - ( edge[u][v].w + dp[v][0] )  ) + ( (edge[u][v].w==1) ? 0 : 1 )

（这个状态转移方程画个图就比较好想，一个节点往上走，可以是它的父亲节点再往上走，也可以是它的父亲节点的其他孩子树，再包括它和它父亲节点相连的那条边）

所以，代码如下：

#include<cstdio>
#include<vector>
#define MAXN 200000+5
using namespace std;
int n,dp[MAXN][2];
struct Edge{
    int u,v,w;
};
vector<Edge> adj[MAXN];
void dfs1(int now,int par)
{
    if(adj[now].empty())
    {
        dp[now][0]=0;
        return;
    }
    for(int i=0;i<adj[now].size();i++)
    {
        Edge edge=adj[now][i];
        int next=edge.v;
        if(next==par) continue;
        dfs1(next,now);
        dp[now][0]+=edge.w+dp[next][0];
    }
}
void dfs2(int now,int par)
{
    if(now==1) dp[now][1]=0;
    for(int i=0;i<adj[now].size();i++)
    {
        Edge edge=adj[now][i];
        int next=edge.v;
        if(next==par) continue;
        dp[next][1] = dp[now][1] + ( dp[now][0] - (edge.w+dp[next][0]) ) + (edge.w?0:1);
        dfs2(next,now);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int from,to;
        scanf("%d%d",&from,&to);
        adj[from].push_back((Edge){from,to,0});
        adj[to].push_back((Edge){to,from,1});
    }
    dfs1(1,0);
    dfs2(1,0);
    vector<int> ans;
    int min=dp[1][0]+dp[1][1];
    for(int i=1;i<=n;i++)
    {
        if(min>dp[i][0]+dp[i][1])
        {
            min=dp[i][0]+dp[i][1];
            ans.clear();
        }
        if(min==dp[i][0]+dp[i][1])
        {
            ans.push_back(i);
        }
    }
    printf("%d
",min);
    for(int i=0;i<ans.size();i++)
    {
        if(i!=0) printf(" ");
        printf("%d",ans[i]);
    }
    printf("
");
}

可以说是第一道大部分都是自己独立完成的树形DP，还是值得庆祝一下的~