树的重心 题解

树的重心

题目描述

给定一颗树,树中包含 n 个结点(编号 1∼n)和 n−1 条无向边。请你找到树的重心,并输出将重心删除后,剩余各个连通块中点数的最大值。

重心定义:重心是指树中的一个结点,如果将这个点删除后,剩余各个连通块中点数的最大值最小,那么这个结点被称为树的重心。

输入格式

第一行包含整数 n,表示树的结点数。

接下来 n−1 行,每行包含两个整数 a 和 b,表示点 a 和点 b 之间存在一条边。

输出格式

输出一个整数 m,表示将重心删除后,剩余各个连通块中点数的最大值。

数据规模与约定

1 ≤ n ≤ 1 0 5 1≤n≤10^5 1≤n≤105

样例输入

复制代码
9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6

样例输出

复制代码
4

思路

树的重心模板题。那么,怎么才能找到某个点(某个重心),将这个点删除后,剩余各个连通块中点数的最大值最小呢?一颗树中,删除某个点后,剩余的各个连通块是该结点子结点所在的各个连通块 和该结点的父结点所在的连通块 。我们可以递归地求该结点的各个子节点所在的连通块大小,并用sum求和,那么父节点所在的连通块大小就是 n − s u m − 1 n-sum-1 n−sum−1 (-1是减去该结点)。至于无向边的存储,可以采用邻接表,将无向边转为两个有向边存储,即存储有向边 ( a , b ) (a,b) (a,b)和 ( b , a ) (b,a) (b,a)。

代码

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxn = 1e5 + 6;
const int maxm = 5e5 + 6;

struct edge
{
    int to; // to为边的指向
};

vector<edge> e[maxn]; // 存储以点i为起点的边

int n;
int _size[maxn];  // u结点的最大子树的结点数
int sum[maxn];    // 以u为根的子树的结点数
bool vis[maxn];   // 结点u是否访问过
int minNum[maxn]; // 删除u后,剩余各个连通块中结点数的最大值

int ans = INT_MAX; // 删除重心后,剩余各个联通块中结点数的最大值

int dfs(int u)
{
    vis[u] = true;    // 标记该点已被访问过
    int cur_size = 0; // u结点的最大子树的结点数
    int cur_sum = 1;  // 以u为根的子树的结点数,初始化时加上该结点本身(即+1)

    for (int i = 0; i < e[u].size(); i++) // 遍历所有以u为起点的边
    {
        int v = e[u][i].to; // v是u的邻接点
        if (vis[v])
            continue;
        int num = dfs(v);              // 递归,并将返回的子树的结点数存储在num中
        cur_size = max(cur_size, num); // 记录最大子树的结点树
        cur_sum += num;                // 累加各个子树的结点数
    }

    _size[u] = cur_size;
    sum[u] = cur_sum;
    minNum[u] = max(cur_size, n - cur_sum);

    ans = min(ans, minNum[u]);

    return cur_sum; // 返回以u为根的子树的结点数
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    cin >> n;
    int a, b;
    // 读入n-1条无向边,并用邻接表存储
    for (int i = 1; i <= n - 1; i++)
    {
        cin >> a >> b;
        e[a].push_back({b});
        e[b].push_back({a});
    }
    dfs(1); // 可以以任意点为起点递归

    cout << ans << '\n';

    return 0;
}
相关推荐
金智维科技官方6 小时前
常见的大模型分类
人工智能·算法·ai·语言模型·数据挖掘
yzzzzzzzzzzzzzzzzz7 小时前
leetcode热题——有效的括号
算法·
崎岖Qiu7 小时前
leetcode1343:大小为K的子数组(定长滑动窗口)
java·算法·leetcode·力扣·滑动窗口
Shun_Tianyou8 小时前
Python Day25 进程与网络编程
开发语言·网络·数据结构·python·算法
Giser探索家9 小时前
什么是2米分辨率卫星影像数据?
大数据·人工智能·数码相机·算法·分类·云计算
jz_ddk10 小时前
[科普] AI加速器架构全景图:从GPU到光计算的算力革命
人工智能·学习·算法·架构
曦月逸霜10 小时前
内部排序算法总结(考研向)
考研·算法·排序算法
仪器科学与传感技术博士11 小时前
Matplotlib库:Python数据可视化的基石,发现它的美
开发语言·人工智能·python·算法·信息可视化·matplotlib·图表可视化
草莓熊Lotso12 小时前
【洛谷题单】--分支结构(二)
c语言·c++·经验分享·其他·刷题
snowfoootball12 小时前
2025 蓝桥杯C/C++国B 部分题解
c语言·c++·笔记·学习·贪心算法·蓝桥杯