首页 > 学院 > 开发设计 > 正文

树的重心

2019-11-10 17:14:05
字体:
来源:转载
供稿:网友

方法:

1、存图遍历,dfs;2、记录两个值,maxsubtree:当前节点的最大子树,d[x]:当前节点的子树和(包括x);3、将当前节点x的maxsubtree再和x的祖先之和比较;4、更新重心和最小连通子树;
#include <iostream>#include <cstdio>#include <cstdlib>#include <cstring>#include <algorithm>#include <cmath>#include <vector>#define inf 1E9#define maxn 10001using namespace std;vector<int> tree[maxn];int n;int zx=0,minzs=inf;int d[maxn];void dfs(int x, int fa) {	d[x]=1;	int maxzs=0;	for(int i=0; i<tree[x].size(); ++i) {		int to=tree[x][i];		if(to!=fa) {			dfs(to,x);			maxzs=max(maxzs,d[to]);			d[x]+=d[to];		}	}	maxzs=max(maxzs,n-d[x]);	if(maxzs<minzs) {		minzs=maxzs;		zx=x;	}}int main() {	scanf("%d", &n);	for(int i=1; i<=n-1; ++i) {		int x,y;		scanf("%d%d", &x, &y);		tree[x].push_back(y);		tree[y].push_back(x);	}	dfs(1,0);	PRintf("%d %d", zx,minzs);	return 0;}
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表