会议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
*/
相关推荐
-dzk-5 小时前
【代码随想录】LC 59.螺旋矩阵 II
c++·线性代数·算法·矩阵·模拟
风筝在晴天搁浅5 小时前
hot100 78.子集
java·算法
Jasmine_llq5 小时前
《P4587 [FJOI2016] 神秘数》
算法·倍增思想·稀疏表(st 表)·前缀和数组(解决静态区间和查询·st表核心实现高效预处理和查询·预处理优化(提前计算所需信息·快速io提升大规模数据读写效率
超级大只老咪6 小时前
快速进制转换
笔记·算法
m0_706653236 小时前
C++编译期数组操作
开发语言·c++·算法
故事和你916 小时前
sdut-Java面向对象-06 继承和多态、抽象类和接口(函数题:10-18题)
java·开发语言·算法·面向对象·基础语法·继承和多态·抽象类和接口
qq_423233906 小时前
C++与Python混合编程实战
开发语言·c++·算法
TracyCoder1237 小时前
LeetCode Hot100(19/100)——206. 反转链表
算法·leetcode
m0_715575347 小时前
分布式任务调度系统
开发语言·c++·算法