CodeForces 219 D. Choosing Capital for Treeland （树形DP）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/imzxww/article/details/81775740

http://codeforces.com/problemset/problem/219/D

题意：

给一个n节点的有向无环图，要找一个这样的点：该点到其它n-1个点要逆转的道路最少，

如果有多个这样的点，则升序输出所有的点

思路：

我们可以把将边的权值方向化，正向为1，反向为0，这样就是求遍历整张图后那些点的权值最大，我们可以从下往上更新到每个子树的权值，再从上往下更新到父亲节点的权值，

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=2e5+10;
struct node
{
    int v,w;
};
vector<node>e[maxn];
int dpf[maxn];
int dps[maxn];
void dfs(int x,int pre)
{
    int k=e[x].size();
    for(int i=0;i<k;i++)
    {
        int v=e[x][i].v;
        int w=e[x][i].w;
        if(v==pre)continue;
        dfs(v,x);
        dps[x]+=w;
        dps[x]+=dps[v];
    }
}
void dfss(int x,int pre)
{
    int k=e[x].size();
    for(int i=0;i<k;i++)
    {
        int v=e[x][i].v;
        int w=e[x][i].w;
        if(v==pre)continue;
        dpf[v]=dpf[x]+dps[x]-dps[v]+(w?0:1)-w;
        dfss(v,x);
    }
}
struct Node
{
    int a,id;
}ans[maxn];
bool cmp(Node a,Node b)
{
    if(a.a!=b.a)return a.a>b.a;
    return a.id<b.id;
}
int main()
{
    int n;
    scanf("%d",&n);
    int u,v;
    node q;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        q.v=v,q.w=1;
        e[u].push_back(q);
        q.v=u,q.w=0;
        e[v].push_back(q);
    }
    dfs(1,-1);
    dfss(1,-1);
    for(int i=1;i<=n;i++)
    {
        ans[i].a=dps[i]+dpf[i];
        ans[i].id=i;
    }
    sort(ans+1,ans+1+n,cmp);
    printf("%d\n",n-1-ans[1].a);
    int p=0;
    for(int i=1;i<=n;i++)
    {
        if(ans[i].a==ans[i-1].a||i==1)
        {
            if(p++)printf(" ");
            printf("%d",ans[i].id);
        }
        else break;
    }
    printf("\n");
    return 0;
}