二叉树-医院设置

题目描述

设有一棵二叉树，如图：

其中，圈中的数字表示结点中居民的人口。圈边上数字表示结点编号，现在要求在某个结点上建立一个医院，使所有居民所走的路程之和为最小，同时约定，相邻接点之间的距离为 11。如上图中，若医院建在1 处，则距离和 =4+12+2×20+2×40=136；若医院建在 33 处，则距离和 =4×2+13+20+40=81。

输入格式

第一行一个整数 nn，表示树的结点数。

接下来的 nn 行每行描述了一个结点的状况，包含三个整数 w, u, vw,u,v，其中 ww 为居民人口数，uu 为左链接（为 00 表示无链接），vv 为右链接（为 00 表示无链接）。

输出格式

一个整数，表示最小距离和。

输入输出样例

输入
5
13 2 3
4 0 0
12 4 5
20 0 0
40 0 0
输出
81

*** Floyd法 ***

#include<cstdio>
using namespace std;
int a[101],g[101][101];//采用邻接矩阵法 
int main(void){
	int n,l,r;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		g[i][j]=1000000;
	for(int i=1;i<=n;i++)
	{
		g[i][i]=0;
		scanf("%d%d%d",&a[i],&l,&r);
		if(l>0)g[i][l]=g[l][i]=1;
		if(r>0)g[i][r]=g[r][i]=1;
	}
	for(int i=1;i<=n;i++)//用Floyed求任意两节点之间的最短路径
	{//头节点 
		for(int j=1;j<=n;j++)//尾节点 
		{
			if(i!=j)
			{
				for(int k=1;k<=n;k++)//中间结点 
					if(i!=k&&j!=k&&g[i][k]+g[k][j]<g[i][j])
						g[i][j]=g[i][k]+g[k][j];
			}
		}
	 } 
	 int min=0x7fffffff;
	 int total;
	 for(int i=1;i<=n;i++){
	 	total=0;
	 	for(int j=1;j<=n;j++)
	 		total+=g[i][j]*a[j];
		if(total<min)	min=total;
	 }
	 printf("%d",min);
	 return 0;
}

*** bfs ***

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
int g[101][101]={0};//邻接矩阵存储 
int num[101]={0};
int ans=1<<30; 
int n,l,r; 
int v[101]={0};
struct node{
	int u,step;
};
int bfs(int x){//bfs找当前点x为医院设置点是的 
	memset(v,0,sizeof(v));
	queue<node> q;
	v[x]=1;
	q.push((node){x,0});
	int sum=0;
	while(!q.empty()){
		node now=q.front();
		q.pop();
		for(int i=1;i<=n;i++)
			if(g[now.u][i]&&!v[i]){
				node next={i,now.step+1};
				sum+=num[i]*next.step;
				v[i]=1;
				q.push(next);
			}
	}
	return sum;
}
int main(void){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)//矩阵初始化
	{
		scanf("%d%d%d",&num[i],&l,&r);
		if(l)g[l][i]=g[i][l]=1;//记住对称！
		if(r)g[i][r]=g[r][i]=1; 
	 } 
	 for(int i=1;i<=n;i++)
	 	ans=min(ans,bfs(i));
	printf("%d",ans);
	return 0;
}