SGU 134 Centroid【树的重心】

最新推荐文章于 2017-10-13 16:50:01 发布

wwwiskey

最新推荐文章于 2017-10-13 16:50:01 发布

阅读量1.1k

点赞数

分类专栏： ACM 数据结构题解

本文链接：https://blog.csdn.net/yang_7_46/article/details/9961025

版权

ACM 同时被 3 个专栏收录

116 篇文章 0 订阅

订阅专栏

题解

89 篇文章 0 订阅

订阅专栏

数据结构

42 篇文章 0 订阅

订阅专栏

树的重心。递归求解即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define N 16100
vector<int> g[N];
int n, cur, s[N], f[N];
void getroot(int now, int fa) {
    s[now] = 1;
    f[now] = 0;
    int u;
    for (int i=0; i<g[now].size(); i++)
        if ((u = g[now][i]) != fa) {
            getroot(u, now);
            s[now] += s[u];
            f[now] = max(f[now], s[u]);
        }
    f[now] = max(f[now], n-s[now]);
    if (f[now] < f[cur]) cur = now;
}
int main() {
    int a, b;
    scanf("%d", &n);
    for (int i=1; i<n; i++) {
        scanf("%d%d", &a, &b);
        g[a].push_back(b); g[b].push_back(a);
    }
    f[cur] = n;
    getroot(1, cur=0);
    vector<int> ans;
    for (int i=1; i<=n; i++) if (f[i] == f[cur]) ans.push_back(i);
    sort(ans.begin(), ans.end());
    printf("%d %d\n", f[cur], ans.size());
    printf("%d", ans[0]);
    for (int i=1; i<ans.size(); i++) printf(" %d", ans[i]);
    return 0;
}