会议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
*/
相关推荐
长安er21 小时前
LeetCode215/347/295 堆相关理论与题目
java·数据结构·算法·leetcode·
元亓亓亓21 小时前
LeetCode热题100--62. 不同路径--中等
算法·leetcode·职场和发展
小白菜又菜21 小时前
Leetcode 1925. Count Square Sum Triples
算法·leetcode
粉红色回忆21 小时前
用链表实现了简单版本的malloc/free函数
数据结构·c++
登山人在路上1 天前
Nginx三种会话保持算法对比
算法·哈希算法·散列表
写代码的小球1 天前
C++计算器(学生版)
c++·算法
AI科技星1 天前
张祥前统一场论宇宙大统一方程的求导验证
服务器·人工智能·科技·线性代数·算法·生活
k***92161 天前
【C++】继承和多态扩展学习
java·c++·学习
序属秋秋秋1 天前
《Linux系统编程之进程控制》【进程等待】
linux·c语言·c++·进程·系统编程·进程控制·进程等待
Fuly10241 天前
大模型剪枝(Pruning)技术简介
算法·机器学习·剪枝