[树 最短路]医院设置

今天给大家讲医院设置这道题目

题目传送门 洛谷

思路

这个题目读完题之后发现题目意思是在一棵二叉树中在其中的某一个点上建一座医院,使得每个节点到他的距离最小。
既然要使一棵树上的距离最小,那我就想到了最短路算法。
我的想法是 我们先运用floyd最短路算法求出树上各个点之间的最小距离,再暴力枚举出在每一个节点上都分别建一个医院的最小距离,最后再取一个最小值即可。
由于这个题目数据范围比较小,所以floyd的O(3)算法以及暴力枚举的O(2)时间复杂度对于这个题目来说,是不成任何问题的。

代码

#include<bits/stdc++.h>
using namespace std;
int a[105],g[105][105];
int n,l,r,minn,total;
int main()
{
	scanf("%d",&n);
	//输入 
	memset(g,0x3f,sizeof(g));//由于算最小距离,所以给两个点之间的距离赋一个极大值 
	for(int i=1;i<=n;i++)
	{
		g[i][i]=0;
		cin>>a[i]>>l>>r;
		if(l>0)
		g[i][l]=g[l][i]=1;
		if(r>0)
		g[i][r]=g[r][i]=1; 
	}
	//floyd
	for(int k=1;k<=n;k++)
	{
		for(int i=1;i<=n;i++)
		{
			if(i!=k)
			{
				for(int j=1;j<=n;j++)
				{
					if(i!=j&&k!=j&&g[i][k]+g[k][j]<g[i][j])
					g[i][j]=g[i][k]+g[k][j];
				}
			}
		}
	}
	//暴力枚举医院建在每一个节点上的最小距离 
	minn=INT_MAX;
	for(int i=1;i<=n;i++)
	{
		total=0;
		for(int j=1;j<=n;j++)
		{
		    total+=g[i][j]*a[j];
		}
		minn=min(total,minn);
	}
	//输出 
	cout<<minn<<endl;
	return 0;
}

完结撒花