树的重心 ← STL及链式前向星两种实现

【题目来源】
https://www.acwing.com/problem/content/848/

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

【输入格式】
第一行包含整数 n，表示树的结点数。
接下来 n−1 行，每行包含两个整数 a 和 b，表示点 a 和点 b 之间存在一条边。

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

【数据范围】
1≤n≤10^5

【输入输出样例】
输入样例：
9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6

输出样例：
4

【算法代码一：STL vector存树】
利用STL中的vector存“树”可参考：https://blog.csdn.net/hnjzsyjyj/article/details/108655516

#include<bits/stdc++.h> 
using namespace std;
 
const int N=1e5+5; 
vector<int> G[N];
int n;
bool st[N];
int ans=N; 

int dfs(int u){ //以u为根的子树中结点的数量
    st[u]=true;
	int sum=1;
    int res=0; //删除某个结点后，最大连通块中结点的数量
    for(auto i:G[u]){ //C++11标准
        if(!st[i]){
            int s=dfs(i);
            res=max(res,s);
            sum+=s;
        }
    } 
    res=max(res,n-sum); 
    ans=min(res,ans);
    return sum;
}
 
int main(){
    cin>>n;
    int x,y;
    for(int i=1;i<n;i++){
        cin>>x>>y;
        G[x].push_back(y);
        G[y].push_back(x);
    }
    dfs(1);
    cout<<ans<<endl;
    return 0;
}


/*
in:
9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6

out:
4
*/

【算法代码二：链式前向星存树】
链式前向星可参考：https://blog.csdn.net/hnjzsyjyj/article/details/139369904

/* 链式前向星存图
val[idx] 表示第 idx 条边的权值。
h[a]：存储单链表表头结点 a 的编号
e[idx]：存储结点 idx 的值
ne[idx]：存储结点 idx 的下一个结点的编号
*/
 
#include <bits/stdc++.h>
using namespace std;
 
const int N=1e5+5;
const int M=N<<1;
int n;
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;
    int 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() {
	memset(h,-1,sizeof(h));
	cin>>n;	
	int x,y;
	for(int i=1; i<n; i++) { //undirected graph
		cin>>x>>y;
		add(x,y);
		add(y,x);
	} 
	dfs(1);	
	cout<<ans<<endl; 
	return 0;
}


/*
in:
9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6

out:
4
*/

【参考文献】
https://blog.csdn.net/hnjzsyjyj/article/details/139369904
https://blog.csdn.net/hnjzsyjyj/article/details/108655516
https://blog.csdn.net/qq_47783057/article/details/116195481
https://www.acwing.com/solution/content/13513/
https://www.acwing.com/solution/content/4917/