POJ3107 Godfather (树形DP)

题意:求树的重心

题解:先跑一遍dfs 预处理出这种遍历方式每个节点的儿子(含自己)的数

　　 再跑一遍 每个点的值就是他所有儿子中取一个最大值 再和它父亲这个方向比较一下

　　 又被卡常了 vector一直tle 需要用前向星....

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <vector>
#include <string.h>
using namespace std;

int read()
{
    int x=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x*f;
}

struct node
{
    int no, to, nex;
}E[];

int n, rt;
int du[];
int head[];
int son[];
int dp[];

int dfs1(int x, int fa)
{
    son[x] = ;
    int c = head[x];
    for(int i = c; i; i = E[i].nex)
    {
        int c = E[i].to;
        if(c == fa) continue;

        son[x] += dfs1(c, x);
    }
    return son[x];
}

void dfs2(int x, int fa)
{
    dp[x] = n - son[x];
    int c = head[x];
    for(int i = c; i; i = E[i].nex)
    {
        int c = E[i].to;
        if(c == fa) continue;

        dp[x] = max(dp[x], son[c]);
        dfs2(c, x);
    }
}

int main()
{
    while(~scanf("%d", &n))
    {
        for(int i = ; i <= n; i++) du[i] = son[i] = dp[i] = head[i] = ;

        int cn = ;
        for(int i = ; i < n; i++)
        {
            int u, v;
            u = read(); v = read();
            E[++cn].no = u;
            E[cn].to = v;
            E[cn].nex = head[u];
            head[u] = cn;

            E[++cn].no = v;
            E[cn].to = u;
            E[cn].nex = head[v];
            head[v] = cn;
            du[u]++;
            du[v]++;
        }

        for(int i = ; i <= n; i++)
            if(du[i] == )
            {
                rt = i;
                break;
            }

        int o = dfs1(rt, -);
        dfs2(rt, -);

        int ans = n + ;
        int cnt = ;
        for(int i = ; i <= n; i++) ans = min(ans, dp[i]);
        for(int i = ; i <= n; i++)
            if(dp[i] == ans) cnt++;

        for(int i = ; i <= n; i++)
        {
            if(dp[i] == ans)
            {
                if(cnt == )
                {
                    printf("%d\n", i);
                    break;
                }
                else printf("%d ", i);
                cnt--;
            }
        }
    }
    return ;
}