树形dp&&树的重心（D - Godfather POJ - 3107）

题目链接：https://cn.vjudge.net/contest/277955#problem/D

题目大意：求树的重心（树的重心指的是树上的某一个点，删掉之后形成的多棵树中节点数最大值最小）。

具体思路：对于每一个点，我们求出以当前的点为根的根数的节点个数， 然后在求树的重心的时候，一共有两种情况，一种树去除该点后这个点的子节点中存在所求的最大值，还有一种情况是这个点往上会求出最大值，往上的最大值就是（n-dp[rt][0]）.

AC代码：

#include<iostream>
#include<cmath>
#include<stack>
#include<stdio.h>
#include<algorithm>
#include<queue>
#include<vector>
#include<cstring>
using namespace std;
# define inf 0x3f3f3f3f
# define ll long long
const int maxn = 2e5+100;
struct node
{
    int nex;
    int to;
} edge[maxn];
int num,head[maxn],dp[maxn][3],father[maxn];
int sto[maxn],minn,n;
void init()
{
    minn=inf;
    num=0;
    memset(head,-1,sizeof(head));
    memset(dp,0,sizeof(dp));
}
void addedge(int fr,int to)
{
    edge[num].to=to;
    edge[num].nex=head[fr];
    head[fr]=num++;
}
void dfs1(int fr,int rt)
{
    dp[fr][0]=1;
    for(int i=head[fr]; i!=-1; i=edge[i].nex)
    {
        int to=edge[i].to;
        if(to==rt)
            continue;
        dfs1(to,fr);
        dp[fr][0]+=dp[to][0];
    }
}
void dfs2(int fr,int rt)
{
    dp[fr][1]=n-dp[fr][0];
    for(int i=head[fr]; i!=-1; i=edge[i].nex)
    {
        int to=edge[i].to;
        if(to==rt)
            continue;
        dfs2(to,fr);
        dp[fr][1]=max(dp[fr][1],dp[to][0]);
    }
    minn=min(minn,dp[fr][1]);
}
int main()
{
    init();
    scanf("%d",&n);
    int t1,t2;
    for(int i=2; i<=n; i++)
    {
        scanf("%d %d",&t1,&t2);
        addedge(t1,t2);
        addedge(t2,t1);
    }
    dfs1(1,-1);
    dfs2(1,-1);
    int flag=1;
    for(int i=1; i<=n; i++)
    {
        if(dp[i][1]==minn)
        {
            if(flag)
            {
                printf("%d",i);
                flag=0;
            }
            else
                printf(" %d",i);
        }
    }
    printf("\n");
    return 0;
}