codeforces 219D 【树形DP】

题目链接：点击打开链接

题意：给出一个有n个结点的有向无环图，问找到一点使能够逆转最少的边到达剩下n-1个点，并且输出所有这样的点

题解：

设dp[i][0]为第i个结点的子树中逆转边的数目，dp[i][1]为第i个结点的除子树外剩下要逆转多少条边。

可以dfs两次，第一次求子树逆转边，第二次求除子树外的逆转边。

#pragma comment(linker, "/STACK:102400000,102400000")
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<cmath>
#include<stdlib.h>
#include <string.h>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<time.h>
#include<sstream>
using namespace std;
#define MAX_N 200005
#define inf 0x7f7f7f7f
#define LL long long
#define ull unsigned long long
const LL INF = 1e18;
//const double Pi = acos(-1);
const int mod = 1e4+7;
typedef pair<int, int>P;

vector<int>G[MAX_N];
set<P>st;
int dp[MAX_N][2];
/*
dp[i][0]该结点子树有多少条边要改，
dp[i][1]该结点的父节点有多少条边要改
*/
void dfs1(int u, int pre)
{
    dp[u][0] = dp[u][1] = 0;
    for(int i=0; i<G[u].size(); i++) {
        int v = G[u][i];
        if(v == pre)
            continue;
        dfs1(v, u);
        if(st.find(P(u, v)) == st.end())//如果u不能到v则需要逆转边
            dp[u][0]++;
        dp[u][0] += dp[v][0];
    }
}
void dfs2(int u, int pre)
{
    for(int i=0; i<G[u].size(); i++) {
        int v = G[u][i];
        if(v == pre)
            continue;
        dp[v][1] = dp[u][0] - dp[v][0] + dp[u][1];//(dp[u][0]-dp[v][0])就是u的除v结点外的其他子树逆转边
        if(st.find(P(u, v)) != st.end())//还需判断u和v之间边的方向
            dp[v][1]++;
        else
            dp[v][1]--;
        dfs2(v, u);
    }
}
int main()
{
    int n;
    cin >> n;
    for(int i=1; i<n; i++) {
        int a, b;
        scanf("%d%d", &a, &b);
        st.insert(P(a, b));
        G[a].push_back(b);
        G[b].push_back(a);
    }
    dfs1(1, 0);
    dfs2(1, 0);
    int mn = inf;
    for(int i=1; i<=n; i++)
        mn = min(mn, dp[i][0]+dp[i][1]);
    cout << mn << endl;
    for(int i=1; i<=n; i++) {
        if(dp[i][0]+dp[i][1] == mn)
            printf("%d ", i);
    }
}