codeforces 219D 树形dp

题目大意：

根据n个城市，设置n-1条有向边，希望设定一个中心城市，能通过修改一些道路的方向到达任何一座城市，希望修改的道路数量最少

输出这个最少的道路数量，并且把所有符合的可作为中心的城市编号一个个输出来

 

开始也实在想不出，根据树形dp，首先要确定一棵树，可是这里的边乱七八糟的，没法确定根

后来看了别人的思路，原来还是自己太死脑筋了，根本不需要确定根，可以把任意一个城市作为根来解决题目，这里假定1为根

把所有的有向边看作无向边，但要记录那条边是真实存在的，哪条是自己加的

用dp[i]表示 i 点到达其对应的子树中的所有点需要修改的边的数量

用一次dfs得到dp[i]

然后网上的大牛们都说2次dfs，后一次因为没写清楚，自己也不想看代码，就自己想了，可就是没办法写出dfs，最后发现直接就可以循环做。。。而且更好理解

做法：

len[i]记录 i 到 root 这里定义root为 1 长度，也就是i 到 1 之间有多少条边

change[i] 表示这 len[i]条边中有多少条改变了

这两个值可以在第一次dfs中实现

然后对于每一个点 i 来说，其实为了让 i 能够到达任何一点，只要改动 i 到 1 这len[i]条边即可，画个图很容易发现

1作为树根，i 到 1 的过程经过的都是一棵棵子树的根，能到达这些子树的根，必能到达子树中的任何一点，所以希望能够到达 1 即可

用rec[i]记录以 i 为中心城市需要改动的边

rec[i] = dp[1] - change[i] + (len[i] - change[i]) //减第一个change[i]是把原来改过的边改回来，然后+后面的是因为为了 i 到1，需要将那些

方向正常的 len[i] - change[i]倒转

 

最后通过rec[i]得到答案即可= =

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int N = 200005;

int first[N] , k , len[N] , change[N] , dp[N] , rec[N];

struct Edge{
    int y , next;
    bool flag;
}e[N<<1];

void add_edge(int x , int y , bool flag)
{
    e[k].y = y , e[k].next = first[x] , e[k].flag = flag;
    first[x] = k++;
}

void dfs(int u , int fa)
{
    for(int i = first[u] ; i!=-1 ; i=e[i].next){
        int v = e[i].y;
        if(v == fa) continue;
        len[v] = len[u]+1;
        change[v] = change[u];
        if(!e[i].flag) change[v]++;
        dfs(v , u);
        if(!e[i].flag){
            dp[u] += dp[v]+1;
        }
        else{
            dp[u] += dp[v];
        }
    }
}

int main()
{
   // freopen("a.in" , "r" , stdin);
    int n , a , b;
    while(scanf("%d" , &n) == 1)
    {
        memset(first , -1 , sizeof(first));
        k = 0;
        for(int i=1 ; i<n ; i++){
            scanf("%d%d" , &a , &b);
            add_edge(a , b , true);
            add_edge(b , a , false);
        }
        memset(dp , 0 , sizeof(dp));
        len[1] = 0 , change[1] = 0;
        dfs(1 , -1);

        int minn = 200000000;

        for(int i=1 ; i<=n ; i++){
            rec[i] = dp[1] + len[i] - 2*change[i];
           // cout<<"i: "<<i<<" "<<len[i]<<" "<<change[i]<<" "<<rec[i]<<endl;
            minn = min(minn , rec[i]);
        }
        printf("%d\n" , minn);
        int num = 0;
        for(int i=1 ; i<=n ; i++){
            if(minn == rec[i]){
                if(num == 0) printf("%d" , i);
                else printf(" %d" , i);
                num ++;
            }
        }
        printf("\n");
    }
    return 0;
}