SGU 134 Centroid(树形dp)

题目链接：SGU 134 Centroid

树形dp。

要找一个点所有子节点(包括这个点的父节点)中具有最大节点数的子节点有两个方向，一个是以这个这个点的所有真正子节点为根的树，另一个是以这个点的父节点为根的树。

这个点父节点那棵树的节点数 = n - （这个点所有子节点的节点数 + 这个点）。

#include <iostream>
#include <stdio.h>
#include <vector>
#include <cstring>
#include <algorithm>

using namespace std;

const int MAX_N = 16000 + 100;
const int MAX_M = MAX_N << 1;
const int INF = (1 << 30);

struct Edge
{
    int v, next;
};
Edge e[MAX_M];
vector <int> vec;
int d[MAX_N], head[MAX_N];
int n, u, v, ans, cnt;

void addEdge(int u, int v)
{
    e[cnt].v = v;
    e[cnt].next = head[u];
    head[u] = cnt;
    cnt++;
}
void dp(int u, int fa)
{
    d[u] = 1;
    int temp = 0;
    int v;
    for(int i = head[u]; i != -1; i = e[i].next)
    {
        v = e[i].v;
        if(v == fa) //添加双向边的意义所在
            continue;
        dp(v, u);
        d[u] += d[v];
        temp = max(temp, d[v]);//从子节点中寻找
    }
    temp = max(temp, n - d[u]);//父节点含节点数与子节点中含最大节点数的那个比较大小
    if(temp < ans)
    {
        ans = temp;
        vec.clear();
        vec.push_back(u);
    }
    else if(temp == ans)
        vec.push_back(u);
}
int main()
{
    scanf("%d", &n);
    memset(head, -1, sizeof(head));
    cnt = 0;
    ans = INF;
    for(int i = 1; i < n; i++)
    {
        scanf("%d%d", &v, &u);
        addEdge(u, v);
        addEdge(v, u);
    }
    dp(1, -1);
    printf("%d %d\n", ans, vec.size());
    sort(vec.begin(), vec.end());
    for(int i = 0; i < vec.size() - 1; i++)
        printf("%d ", vec[i]);
    printf("%d\n", vec[vec.size() - 1]);
    return 0;
}