第一届ACC决赛 - C.翻转树边【树形DP】

Date：2022.03.27
题意描述：
给定一个 n 个节点的树。
节点编号为 1∼n。
树中的 n−1 条边均为单向边。
现在，我们需要选取一个节点作为中心点，并希望从中心点出发可以到达其他所有节点。
但是，由于树中的边均为单向边，所以在选定中心点后，可能无法从中心点出发到达其他所有节点。
为此，我们需要翻转一些边的方向，从而使得所选中心点可以到达其他所有节点。
我们希望选定中心点后，所需翻转方向的边的数量尽可能少。
请你确定哪些点可以选定为中心点，并输出所需的最少翻转边数量。
输入格式
第一行包含整数 n。
接下来 n−1 行，每行包含两个整数 a,b，表示存在一条从 a 到 b 的单向边。
输出格式
第一行输出一个整数，表示所需的最少翻转边数量。
第二行以升序顺序输出所有可选中心点（即所需翻转边数量最少的中心点）的编号。
数据范围
前三个测试点满足 2≤n≤5。
所有测试点满足 2≤n≤2×105，1≤a,b≤n，a≠b。
输入样例1：
3
2 1
2 3
输出样例1：
0
2
输入样例2：
4
1 4
2 4
3 4
输出样例2：
2
1 2 3

思路：一眼树形dp，想了想又觉得可能是结论，画了图墨迹了一会发现还是树形dp，还算比较裸。显然给定的正权花费0，负权边花费1。
$f[i]:$从$i$号点开始出发，单向到达所有点（可间接）所需的权值和。
因此求出所有的$f[i]$，最小的值即为最少翻转边数量。
树形dp的过程有两种，因为每次都是两个点之间的关系：设结点i、j为紧邻的父、子结点。
①$f[u]+=(f[j]+g[i][j]);$这个很好理解。
②$f[j]+=(f[u]-(f[j]+g[i][j])+g[j][i]);$我们来画个图。

也很好理解。
最后求出最小的，再找到所有$f[i]=minn$的$i$即可。

代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL N = 2e5+10,M=4e5+10,INF=1e18;
LL a[N],n,m,t,k,h[N],e[M],ne[M],w[M],idx;
LL f[N];
void add(LL a, LL b, LL c)  // 添加一条边a->b，边权为c
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}
void dfs_down(LL u,LL baba)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        LL j=e[i];
        if(j==baba) continue;
        dfs_down(j,u);
        f[u]+=(f[j]+w[i]);
    }

}
void dfs_up(LL u,LL baba)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        LL j=e[i];
        if(j==baba) continue;
        f[j]+=(f[u]-(f[j]+w[i]));
        if(w[i]==0) f[j]++;//反向是1
        dfs_up(j,u);
    }
}
int main() 
{
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n;memset(h,-1,sizeof h);
    for(int i=1;i<=n-1;i++)
    {
        LL x,y;cin>>x>>y;
        add(x,y,0);add(y,x,1);
    }
    dfs_down(1,-1);
    dfs_up(1,-1);
    LL minn=INF;
    for(int i=1;i<=n;i++) minn=min(minn,f[i]);
    cout<<minn<<'\n';
    for(int i=1;i<=n;i++)
        if(f[i]==minn) cout<<i<<' ';
    return 0;
}