首都选址【floyd算法（弗洛伊德）】

某国准备选择新的首都，并对新首都的位置提出要求：即新首都到该国所有其他城市的平均距离最短（当然并不一定要求城市之间直达，间接道路也是允许的，但必须可达）。

已知该国的城市信息，请回答新首都应选择哪一个城市。

输入格式:

首先输入一个正整数T，表示测试数据的组数，然后是T组测试数据。每组测试数据第一行输入2个整数n和m，表示该国的城市总数以及现有的道路总数（1≤n≤100， 0≤m≤n*(n-1)/2 ），为方便起见，我们规定城市编号为1到n。
接下来输入m行数据，每行3个整数u、v、d（1≤u,v≤n，1≤d≤100），分别代表连接某条道路的两个城市编号以及该道路的长度。

输出格式:

对于每组测试，输出新首都的城市编号，若有多个满足条件的城市，选择编号最小的，若没有城市满足新首都的条件，请输出0。

输入样例:

3
4 5
1 2 10
1 3 20
4 3 5
4 1 7
2 4 8
3 1
1 2 10
5 7
1 2 2
3 2 5
2 5 18
4 2 6
1 3 8
3 4 11
1 4 8

输出样例:

4
0
2

弗洛伊德算法： 

1.可以用于求解的问题：多源最短路（求任意两个点间的最短路） 

2.弗洛伊德算法难点：k是中转点； 

3.代码： 

void floyd(){
	for(int k=1;k<=n;k++){
		for(int j=1;j<=n;j++){
			for(int i=1;i<=n;i++)
			{
				a[i][j]=min(a[i][j],a[i][k]+a[k][j]);
			}
		}
	}
}

代码： 

#include<bits/stdc++.h>
using namespace std;
const int N=1005,INF=0x3f3f3f3f3f;
int sum[N],a[N][N];
int n,m;//城市总数 
void floyd(){
	for(int k=1;k<=n;k++){
		for(int j=1;j<=n;j++){
			for(int i=1;i<=n;i++)
			{
				a[i][j]=min(a[i][j],a[i][k]+a[k][j]);
			}
		}
	}
}
int main(){
	int t;
	cin>>t;
	while(t--){
		memset(sum,0,sizeof(sum));
		cin>>n>>m;
		//将图都赋为最大值 
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				if(i==j) a[i][j]=0;
				else a[i][j]=INF;
			}
		}
		
		int u,v,d;
		for(int k=1;k<=m;k++){
			cin>>u>>v>>d;
			a[u][v]=a[v][u]=min(a[u][v],d);
		} 
		floyd();//更新每个点到另一个点的最短距离 
		
		int f=0;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				if(a[i][j]==INF) f=1;//没有路与其相连接
				sum[i]+=a[i][j]; 
			}
		}
		
		int minn=INF,index=0;
		for(int i=1;i<=n;i++)
		{
			if(sum[i]<minn)
			{
				minn=sum[i];
				index=i;
			}
		 } 
		 
		 if(f==1) cout<<"0"<<endl;
		 else{
		 	cout<<index<<endl;
		 }
	}
	return 0;
}