算法板子:树形DP、树的DFS——树的重心

思想:

在这里插入图片描述

代码:

#include <iostream>
#include <cstring>
using namespace std;

const int N = 1e5 + 10;

// vis标记当前节点是否被访问过; vis[1]=true代表编号为1的节点被访问过
bool vis[N];
// h数组为邻接表; h数组上的每个坑位都串了一个单链表; h[1]=3代表编号为1的节点的第一个邻接节点的编号为3
// e[i]代表编号为i的节点的值; ne[i]代表编号为i的节点的下一个节点的编号; idx代表当前可用编号; 注意这里的e和ne数组要开2*N的大小
int h[N], e[2 * N], ne[2 * N], idx;

// res存储去掉各个节点后的最大连通块的最小值; 初始记录为最大值,最大值为最大连通块的点数,就是树中所有的节点数
int n, res = N;

// 值为a的节点和值为b的节点之间存在一条边,在以a打头的单链表上插入头结点b
void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

// 得到以u为根的树的总节点数
int dfs(int u)
{
    vis[u] = true; // 标记u节点已经访问过
    int size = 0; // 代表以u为根的最大子树的节点数; 初始时最大子树节点数为0
    int sum = 1; // 代表以u为根的树的总节点数; 初始时该树的总节点数为1
    // 遍历u节点的邻居; 也就是遍历u的每个儿子
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i]; // 得到u节点的邻居节点,也就是u的儿子
        if (vis[j]) continue; //避免叶子结点找到的邻居节点为父节点
        int s = dfs(j); // 得到以某个儿子节点为根的树的总节点数; 也就是某个子树的节点数
        size = max(size, s); // 更新最大子树节点数
        sum += s; // 累加所有子树的节点数
    }
    
    // 如果删掉了u这个点,那么整个树可以分成两部分,u的所有子树和u的上面的剩余部分
    // u的所有子树的最大节点数为size,剩余部分为整棵树的总节点数减去以u为根的树的节点数,求max后得到删去u后连通块中的最大节点数
    // 再和删去其他节点的答案作比较,更新最小的最大连通块的点数
    res = min(res, max(size, n - sum));
    
    return sum;
}

int main()
{
    cin >> n;
    
    // 首先要初始化邻接表h
    memset(h, -1, sizeof h);
    // 将输入的点构成邻接表
    for (int i = 0; i < n - 1; i ++ )
    {
        int a, b;
        cin >> a >> b;
        // 比如a=1 b=3,代表值为1的节点和值为3的节点之间存在边; 那么1的单链表后面要串个3,3的单链表后面要串个1
        add(a, b); add(b, a);
    }
    
    // 从树中的任意一个节点开始dfs都可以; 比如也可以是dfs(n-1)
    dfs(1);
    
    cout << res << endl;
    
    return 0;
}
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值