Description
岛要修路,这个岛上有n个城市,要求修完路后,各城市之间可以相互到达,且修的总路程最短,求所修路中的最长的路段
Input
第一行为用例组数T,每组用例第一行为距离矩阵的行列数n,之后为一n*n矩阵表示各城市之间距离
Output
对于每组用例,输出总路程最短时所修路中最长的路段
Sample Input
1
3
0 990 692
990 0 179
692 179 0
Sample Output
692
Solution
最小生成树问题,prim算法
Code
#include<cstdio>
#include<iostream>
using namespace std;
#define INF 100000000
#define maxn 505
long cost[maxn][maxn];
long dis[maxn];//用以记录当前树到各个顶点的最小距离
bool used[maxn];//标记数组,没有加入到树中时为false,加入了为true
int T,n;
long prim()
{
for(int i=0;i<n;i++)//初始化
{
dis[i]=cost[0][i];
used[i]=false;
}
used[0]=true;//将第一个顶点加入树中
long ans=-1;
for(int i=0;i<n-1;i++)
{
long temp=INF;
int temp_pos=0;
for(int j=0;j<n;j++)//找最小的边
if(!used[j]&&dis[j]<temp)//找出没有在当前树中且权值最小的点
{
temp=dis[j];
temp_pos=j;
}
ans=ans>temp?ans:temp;//更新所修路中的最长路
used[temp_pos]=true;//将顶点temp-pos加入树中
for(int j=0;j<n;j++)//distan[i]中始终放生成树到顶点i的最小权值
{
if(!used[j]&&dis[j]>cost[temp_pos][j])
dis[j]=cost[temp_pos][j];
}
}
return ans;
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
scanf("%ld",&cost[i][j]);
printf("%ld\n",prim());
}
return 0;
}