树的重心1——树的重心

最新推荐文章于 2023-11-28 20:47:05 发布

gaomuchen_120309

最新推荐文章于 2023-11-28 20:47:05 发布

阅读量227

点赞数

文章标签：深度优先算法 c++ 数据结构图论

本文链接：https://blog.csdn.net/gaomuchen_120309/article/details/131013541

版权

题目：

树的重心1

时间限制：1秒内存限制：128M

题目描述

对于多叉树(多叉树包含二叉树)而言，如我们删除一个节点x后，原来的树可能会分成若干个不相连的部分，其中每个部分都是一棵子树。设 $ma x$ _ $p a r t (x)$ 表示在删除节点x的所有子树中，最大的一棵树的大小。使 $ma x$ _ $p a r t$ 的函数取到最小值的节点p就被称为整棵树的重心。
现在给你一颗多叉树，请你求出这棵树的重心(保证树的重心唯一)。

输入描述

第一行，两个整数 $n$ , $s$ ,其中 $n (1 <= n <= 100000)$ 表示树节点的个数， $s$ 表示根节点的编号。
接下来的 $n - 1$ 行中，每行有两个整数 $x$ , $y$ ，表示 $x$ 和 $y$ 有一条边。

输出描述

输出占一行，1个数，输出树的重心所在的节点号和最大子树的节点个数。

样例

输入

输出

1 4

思路及代码

解题思路

输入n和s。

scanf("%d %d",&n,&s);

循环输入树的每条边，并构建树

for(int i=1;i<n;i++){
	int a,b;
	scanf("%d %d",&a,&b);
	add(a,b);
	add(b,a);
}

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

循环从每一个节点进行DFS搜索

for(int i=1;i<=n;i++){
	dfs(i);
}

4.DFS搜索时，先把存最大子树大小的变量maxpart出始化为0，把起点x用vis数组标记掉，把x的子树大小设为1，并存进sizes数组里。再遍历所有邻接点，如果它没有被标记掉，用vis数组标记掉，从它再次DFS搜索，然后将x的子树大小加上它的子树大小，用它的子树大小与maxpart比较。用未遍历到的子树大小与maxpart比较。把存最小最大子树大小的变量part与maxpart比较，如果maxpart小，part更新为maxpart，把存树的重心的变量node存下当前节点。

void dfs(int x){
	int maxpart=0;
	vis[x]=1;
	sizes[x]=1;
	if(x<1){
		return ;
	}
	for(int i=head[x];i!=-1;i=ne[i]){
		if(vis[e[i]]==0){
			vis[e[i]]=1;
			dfs(e[i]);
			sizes[x]+=sizes[e[i]];
			maxpart=max(maxpart,sizes[e[i]]);
		}
	}
	maxpart=max(maxpart,n-sizes[x]);
	if(maxpart<part){
		part=maxpart;
		node=x;
	}
}

输出树的重心和最小最大子树的大小

cout<<node<<" "<<part;

AC代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
int n;
int head[200002];
int e[200002];
int ne[200002];
int vis[200002];
int sizes[100002];
int idx=0;
int part=100001,node;
void add(int a,int b){
	e[++idx]=b;
	ne[idx]=head[a];
	head[a]=idx;
}
void dfs(int x){
	int maxpart=0;
	vis[x]=1;
	sizes[x]=1;
	if(x<1){
		return ;
	}
	for(int i=head[x];i!=-1;i=ne[i]){
		if(vis[e[i]]==0){
			vis[e[i]]=1;
			dfs(e[i]);
			sizes[x]+=sizes[e[i]];
			maxpart=max(maxpart,sizes[e[i]]);
		}
	}
	maxpart=max(maxpart,n-sizes[x]);
	if(maxpart<part){
		part=maxpart;
		node=x;
	}
}
int main() {
	memset(head,-1,sizeof head);
	int s;
	scanf("%d %d",&n,&s);
	for(int i=1;i<n;i++){
		int a,b;
		scanf("%d %d",&a,&b);
		add(a,b);
		add(b,a);
	}
	for(int i=1;i<=n;i++){
		dfs(i);
	}
	cout<<node<<" "<<part;
	return 0;
}

知识点：

树的重心的特征

树的重心有以下特征：
1.删除重心后所得的所有子树，节点数不超过原树的1/2，并且让节点数最多的子树节点数尽可能少，一棵树最多有两个重心；
2.树中所有节点到重心的距离之和最小，如果有两个重心，那么他们距离之和相等；
3.两个树通过一条边合并，新的重心在原树两个重心的路径上；
4.树删除或添加一个叶子节点，重心最多只移动一条边；