C++ 树与图的深度优先遍历 || 模版题:树的重心

news/2024/7/15 18:29:05 标签: 深度优先, c++, 图论

树和无向图都可以看成有向图(无向图在添加边的时候添加双向的)
下面是模版,实际使用要根据情况改:

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

const int N = 10010, M = N * 2;

int n;
int h[N], e[M], ne[M], idx;
bool st[N];

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

void dfs(int u)
{
    st[u] = true;
    for(int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if(!st[j])
            dfs(j);
    }
}

int main ()
{
    memset(h, -1, sizeof h);
    
    dfs(1);
}

给定一颗树,树中包含 n
个结点(编号 1∼n
)和 n−1
条无向边。

请你找到树的重心,并输出将重心删除后,剩余各个连通块中点数的最大值。

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

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

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

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

数据范围
1≤n≤105
输入样例
9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6
输出样例:
4

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

const int N = 10010, M = N * 2;

int n, m;
int h[N], e[M], ne[M], idx;
bool st[N];

int ans = N;

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

int dfs(int u) //返回以u为根的子树中点的数量
{
    st[u] = true;
    int sum = 1, res = 0;
    for(int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if(!st[j])
        {
            int s = dfs(j);
            res = max(res, s);
            sum += s;
        }
    }
    res = max(res, n - sum);
    
    ans = min(ans, res);
    
    return sum;
}

int main ()
{
    cin>>n;
    memset(h, -1, sizeof h);
    
    for(int i = 0; i < n - 1; i ++ )
    {
        int a, b;
        cin>>a>>b;
        add(a, b), add(b, a);
    }
    
    dfs(1);
    
    cout<<ans<<endl;
    
    return 0;
}

h[N], e[M], ne[M], idx: 采用邻接表的方式存储树的结构。h 数组用于存储每个节点的邻接链表的头结点,e 数组存储邻接表的节点信息,ne 数组存储邻接表节点间的连接关系,idx 是邻接表的索引。
add 函数:用于添加边到邻接表中,因为树是无向图,所以需要同时添加两个方向的边。
dfs 函数:深度优先搜索,返回以当前节点 u 为根的子树中的节点数量。在搜索的同时,计算去掉当前节点后,剩余部分的最大子树节点数量,并更新 res。递归地计算各个子节点。
main 函数:读入树的边,构建邻接表。调用 dfs(1) 从根节点开始搜索,计算树的重心。输出重心的最大子树节点数量。


http://www.niftyadmin.cn/n/5326968.html

相关文章

【嵌入式学习笔记-02】什么是库文件,静态库的制作和使用,动态库的制作和使用,动态库的动态加载

【嵌入式学习笔记-02】什么是库文件&#xff0c;静态库的制作和使用&#xff0c;动态库的制作和使用&#xff0c;动态库的动态加载 文章目录 什么是库文件&#xff1f;编程模型的发展什么是库文件&#xff1f; 静态库的制作和使用动态库的制作和使用动态库的动态加载 什么是库文…

程序的内存模型\全局区

之前介绍了C语言的一些内容&#xff0c;之后会不定期更新 今天要介绍的C和C&#xff0c;最开始其实差别不算大&#xff0c;在很多地方用法是一致的&#xff0c;但后来制定的标准将两者进行区分&#xff0c;详情可以查找conference C与C最大的不同在于&#xff0c;C偏向于面向过…

mfc140.dll丢失了要怎么解决?多种解决mfc140.dll方法对比

mfc140.dll丢失了怎么办&#xff1f;mfc140.dll丢失的多个解决方法今天就来给大家详细的说说&#xff0c;因为平时实在是太多人咨询这个问题了&#xff0c;都说自己电脑的mfc140.dll找不到了&#xff0c;一些程序没办法运行&#xff0c;所以还是给大家分析一下mfc140.dll。 一.…

AUTOSAR有没有可能被其他的软件框架取代?

摘要: CP(Classic Platform)经过10多年发展已经是非常成熟的框架,经过了众多量产项目考验。模块的功能和API成熟度高。 观点:只要传统主机厂- 供应商的开发体系不变,使用AUTOSAR的现状就不会改变。传统主机厂(特别是欧洲的主机厂)和一些没有软件能力的主机厂会继续大规…

x-cmd pkg | czg - git commit 智能生成工具

目录 简介首次用户功能特点竞品和相关作品进一步探索 简介 czg 源于 commitizen/cz-cli 交互插件中 cz-git 的延伸项目&#xff0c;重新使用 TypeScript 编写的零依赖独立的 Node.js 命令行工具。旨在使用交互友好的方式&#xff0c;辅助用户生成规范的 git commit message 约…

区间预测 | Matlab实现LSSVM-ABKDE的最小二乘支持向量机结合自适应带宽核密度估计多变量回归区间预测

区间预测 | Matlab实现LSSVM-ABKDE的最小二乘支持向量机结合自适应带宽核密度估计多变量回归区间预测 目录 区间预测 | Matlab实现LSSVM-ABKDE的最小二乘支持向量机结合自适应带宽核密度估计多变量回归区间预测效果一览基本介绍程序设计参考资料 效果一览 基本介绍 1.Matlab实现…

【CSCV】划分数据集

参考论文IEEE Xplore Full-Text PDF: 划分数据集时多了一个development set&#xff0c;如下图 先占个坑&#xff0c;看完论文再来填坑

String引用类型

目录 前言&#xff1a; String类&#xff1a; 字符串的比较&#xff1a; 关于String的一些方法&#xff1a; charAt&#xff1a; indexOf&#xff1a; lastIndexOf: valueOf: parselnt: toLowerCase和toUpperCase: toCharArray: format: replace: replace…