会议dfs树

会议

题目描述

有一个村庄居住着 n 个村民,有 n-1 条路径使得这 n 个村民的家联通,每条路径的长度都为 1。

现在村长希望在某个村民家中召开一场会议,村长希望所有村民到会议地点的距离之和最小,那么村长应该要把会议地点设置在哪个村民的家中,并且这个距离总和最小是多少?

若有多个节点都满足条件,则选择节点编号最小的那个点。

输入格式

第一行,一个数 n,表示有 n 个村民。

接下来 n-1 行,每行两个数字 a 和 b,表示村民 a 的家和村民 b 的家之间存在一条路径。

输出格式

一行输出两个数字 x 和 y。

x 表示村长将会在哪个村民家中举办会议。

y 表示距离之和的最小值。

样例输入1

4

1 2

2 3

3 4

样例输出1

2 4

注释说明

对于70%数据,n<=1000

对于100%数据,n<=50000

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
vector<int>a[50002];
int f[50002],sum[50002],n,ans=INT_MAX,num;
void dfs(int x){
	f[x]=1;
	for(int i=0;i<a[x].size();i++){
		int to=a[x][i];
		dfs(to);
		f[x]+=f[to];
		sum[x]+=sum[to];
	}
	sum[x]+=f[x]-1;
}
void dfs2(int x){
	
	if(ans>sum[x]){
		ans=sum[x];
		num=x;
	}
	for(int i=0;i<a[x].size();i++){
		int to=a[x][i];
		sum[to]=sum[x]+n-2*f[to];//printf("%d %d %d\n",x,to,sum[to]);
		dfs2(to);
	}
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		if(u>v)swap(u,v);
		a[u].push_back(v);
	}
	dfs(1);
//	for(int i=1;i<=n;i++){
//		printf("%d ",sum[i]);
//	}
	dfs2(1);
	printf("%d %d",num,ans);
}
/*

4
1 2
2 3
3 4
*/
相关推荐
机器视觉知识推荐、就业指导1 小时前
C++设计模式:建造者模式(Builder) 房屋建造案例
c++
Swift社区2 小时前
LeetCode - #139 单词拆分
算法·leetcode·职场和发展
Kent_J_Truman3 小时前
greater<>() 、less<>()及运算符 < 重载在排序和堆中的使用
算法
IT 青年3 小时前
数据结构 (1)基本概念和术语
数据结构·算法
Yang.993 小时前
基于Windows系统用C++做一个点名工具
c++·windows·sql·visual studio code·sqlite3
熬夜学编程的小王3 小时前
【初阶数据结构篇】双向链表的实现(赋源码)
数据结构·c++·链表·双向链表
zz40_3 小时前
C++自己写类 和 运算符重载函数
c++
Dong雨4 小时前
力扣hot100-->栈/单调栈
算法·leetcode·职场和发展
六月的翅膀4 小时前
C++:实例访问静态成员函数和类访问静态成员函数有什么区别
开发语言·c++