题解【acwing】846: 树的重心

题目描述

点击进入题目
给定一颗树，树中包含n个结点（编号1~n）和n-1条无向边。请你找到树的重心，并输出将重心删除后，剩余各个连通块中点数的最大值。重心定义：重心是指树中的一个结点，如果将这个点删除后，剩余各个连通块中点数的最大值最小，那么这个节点被称为树的重心。

思路

1. 用邻接表存储数组
2. 用dfs枚举出结果

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
const int N=1e5+10,M=N*2;

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

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

int dfs(int u)
{
    st[u]=true;
    int size=0,sum=0;
    for(int i=h[u]; i!=-1; i=ne[i])
    {
        int j=e[i];
        if(st[j]) continue;
        
        int s=dfs(j);
        sum+=s;
        size=max(size,s);
    }
    
    size=max(size,n-sum-1);
    ans=min(ans,size);
    return sum+1;
}
h
int main()
{
    memset(h,-1,sizeof(h));
    cin>>n;
    for(int i=0; i<n-1; i++)
    {
        int a,b;
        cin>>a>>b;
        add(a,b);
        add(b,a);
    }
    
    dfs(1);
    cout<<ans;
}