AcWing 846. 树的重心【DFS】【树形DP】

Derrickhang

已于 2022-03-30 00:00:50 修改

阅读量375

点赞数

分类专栏：搜索文章标签：算法 c++ 数据结构搜索

于 2022-03-30 00:00:10 首次发布

本文链接：https://blog.csdn.net/Derrickhang/article/details/123835684

版权

搜索专栏收录该内容

21 篇文章 1 订阅

订阅专栏

AcWing 846. 树的重心

一、题目链接
二、题目分析
- （一）算法标签
- （二）解题思路
三、AC代码
四、其它题解

一、题目链接

AcWing 846. 树的重心

二、题目分析

（一）算法标签

DFS 树形DP

（二）解题思路

三、AC代码

解法一:

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10, M = 2 * N;

int h[N], e[M], ne[M], idx;

int n;
int a, b;
int st[N];
int ans = N;

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

// 求以u为根结点的各个子树的结点的数量
int dfs(int u)
{
    st[u] = true;
    
    int res = 0, sum = 1; 
    for (int i = h[u]; ~i; 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 ++ )
    {
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);   // 无向图 建两条边
    }
    
    dfs(1); // 从任何一个点开始都行
    
    cout << ans << endl;

    return 0;
}