描述
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本(道路是双向的)。现请你编写程序,计算出全省畅通需要的最低成本。
-
输入
-
测试输入包含若干测试用例。第一行一个整数 T (T <= 5) 表示测试用例数量,每个测
试用例的第 1 行给出评估的道路条数 N、村庄数目 M ( 1 <= M < 100,0 <= N <= M *( M-1) /2),随后的 N 行对应村庄间道路的成本,每行给出三个正整数,依次是两个村庄的编号(每对村庄至多出现一次),以及此两村庄间道路的成本( 也是正整数 )。
为简单起见,村庄从 1 到 M 编号。
输出
- 对每个测试用例,在 1 行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出 No solution。 样例输入
-
2 3 3 1 2 1 1 3 2 2 3 4 1 3 2 3 2
样例输出
-
3 No solution
#include<stdio.h>
#include<string.h>
#define max 0x3f3f3f3f
#define N 1010
int bin[N];
int a[N][N],visit[N],low[N];
int find(int x)
{
while(bin[x]!=x)
{
x=bin[x];
}
return x;
}
void merge(int x,int y)
{
int fx,fy;
fx=find(x);
fy=find(y);
if(fx!=fy)
{
bin[fx]=fy;
}
}
int main()
{
int n,m;
int i,j;
int b,c,d,min;
int t;
scanf("%d",&t);
while(t--)
{
memset(visit,0,sizeof(visit));
//memset(low,max,sizeof(low));
memset(a,max,sizeof(a));
scanf("%d%d",&n,&m);
for(i=1;i<=m;++i)
{
bin[i]=i;
}
while(n--)
{
scanf("%d%d%d",&b,&c,&d);
a[b][c]=a[c][b]=d;
merge(b,c);
}
int cnt=0;
for(i=1;i<=m;++i)
{
if(bin[i]!=i) ++cnt;
}
if(cnt<m-1)
{
printf("No solution\n");
continue;
}
visit[1]=1;int pos=1;//第一次给low赋值
for(i=1;i<=m;++i)
{
if(i!=pos) low[i]=a[pos][i];
}
int result=0;
//运行m-1次,因为至少需要m-1次才能把所有的城市连通
for(i=1;i<m;++i)
{
min=max;
for(j=1;j<=m;++j)
{
if(!visit[j]&&min>low[j])
{
min=low[j];
pos=j;
}
}
result+=min;
//printf("%d\n",result);
visit[pos]=1;
for(j=1;j<=m;++j)
{
if(!visit[j]&&low[j]>a[pos][j])
{
low[j]=a[pos][j];
}
}
}
printf("%d\n",result);
}
return 0;
}