题目描述
输入
输出
示例输入
3 2 1 2 1 1 3 1 1 0
示例输出
2 0
提示
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define inf 0x3f3f3f
int mp[150][150],vis[150],dis[150],sum;
int prim(int n)
{
int i,j,k;
memset(vis,0,sizeof(vis));
for(i=1;i<=n;i++)
{
dis[i]=mp[1][i];
}
vis[1]=1;
for(i=2;i<=n;i++)//一次一条边
{
int min=inf;
for(j=1;j<=n;j++)
{
if(min>dis[j]&&!vis[j])//每次找到最小权
{
k=j;
min=dis[j];
}
}
sum+=min;
vis[k]=1;
for(j=1;j<=n;j++)
{
if(mp[k][j]<dis[j]&&!vis[j])//如果k到该点的代价要更大一些那么保留1到该点(原来的)代价,一样可以把所有点连起来
{
dis[j]=mp[k][j];//dis中存的一直是相比各点权值最小的
}
}
}
return sum;
}
int main()
{
int n,m,a,b,c,i,j;
while(~scanf("%d%d",&n,&m))
{ sum=0;
for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
{
if(i==j)
mp[i][j]=0;
else
mp[i][j]=inf;
}
}
for(i=0;i<=m-1;i++)
{
scanf("%d%d%d",&a,&b,&c);
if(mp[a][b]>c)//得到两点间最小代价
mp[a][b]=mp[b][a]=c;
}
int x=prim(n);
if(m<n-1)
printf("0\n");
else
printf("%d\n",x);
}
return 0;
}