AcWing 846.树的重心

#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5+10,M=N*2;
int h[N],e[M],ne[M],idx;//存储邻接表 h 是n个链表 e存储n个节点的值 ne[]存储节点指向
//无向边 每条边要存两遍 就×2
void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;//插入
}
int main()
{
    memset(h,-1,sizeof h);//初始化为-1
}

深度搜索:

 

宽度搜索：

#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5+10,M=N*2;//N 数据范围 M边 ab ba存储
int h[N],e[M],ne[M],idx;//h n个单链表 e 存节点编号 
int n,ans=N;//ans答案
bool st[N];//判断节点是否访问

void add(int a,int b)//将b加到a节点所指向的链表中
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

int dfs(int u)
{
    st[u]=true;//标记已访问
    
    int sum=1,res=0;//sum 节点数 res 连通块最大点数
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];//节点编号值
        if(!st[j])//未被访问
        {
            int s=dfs(j);//dfs s是节点编号为j根的节点数
            res=max(res,s);
            sum+=s;//节点数增加
        }
    }
    
    res=max(res,n-sum);//上部分和res比较
    ans=min(ans,res);
    return sum;
}

int main()
{
    scanf("%d",&n);
    memset(h,-1,sizeof h);//链表初始-1
    for(int i=0;i<n;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    
    dfs(1);
    cout<<ans<<endl;
    return 0;
}