CF 219D Choosing Capital for Treeland 树形DP

首先我们可以将有向图转化为无向图，即正向边权值为0，反向边权值为1。那么接下来求某个点到剩下所有点的距离最小值。首先我们可以选择一个点为根，求出其到剩下所有点的距离。那么接下来选择其它点为跟的情况可以由上面计算的值递推出来。

假设root有son1和son2, root到其他所有点, 都是先到son1, son2, 再接着往下走.那么求son1到所有点的时候, son1到son1子树的计数情况和root的计数情况相同, son1到root之后再走son2子树的计数情况也和root相同. 于是只有son1到root之间一段不同.

如果是root->son1, 那么dp[son1] = dp[root] + 1; 若是son1->root,那么dp[son1] = dp[root] - 1;再dfs一遍即可.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define INF 0x3f3f3f3f
const int maxn=2e5+10;
vector<pair<int,int>> G[maxn];
int sum[maxn];

void dfs1(int u,int fa)
{
    int len=G[u].size();
    for(int i=0;i<len;i++)
    {
        int v=G[u][i].first;
        int w=G[u][i].second;
        if(v==fa) continue;
        if(w) sum[u]++;
        dfs1(v,u);
        sum[u]+=sum[v];
    }
}

void dfs2(int u,int fa)
{
    int len=G[u].size();
    for(int i=0;i<len;i++)
    {
        int v=G[u][i].first;
        int w=G[u][i].second;
        if(v==fa) continue;
        if(w) sum[v]=sum[u]-1;
        else sum[v]=sum[u]+1;
        dfs2(v,u);
    }
}


int main()
{
    int n,u,v;
    while(~scanf("%d",&n))
    {
        memset(sum,0,sizeof sum);
        for(int i=1;i<=n;i++) G[i].clear();
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&u,&v);
            G[u].push_back(make_pair(v,0));
            G[v].push_back(make_pair(u,1));
        }
        dfs1(1,0);
        dfs2(1,0);
        int ans=INF;
        for(int i=1;i<=n;i++)
            ans=min(ans,sum[i]);
        printf("%d\n",ans);
        int flag=1;
        for(int i=1;i<=n;i++)
        {
            if(sum[i]==ans)
            {
                if(flag){printf("%d",i);flag=0;}
                else printf(" %d",i);
            }
        }
        puts("");
    }
    return 0;
}