poj 1679

次小生成树练习

判断有无相同最小边。#include <iostream>
#include <cstdio>
#include <cstring>
#define MAX 1e9
using namespace std;
int n,m;
int _map[105][105];
int vis[105],dis[105];
void init()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
_map[i][j]=i==j?0:MAX;

for(int i=1;i<=n;i++)
dis[i]=MAX;

}

void prim()
{
int sum=0;
for(int i=1;i<=n;i++)
dis[i]=_map[1][i];

memset(vis,0,sizeof vis);
vis[1]=1;
for(int i=1;i<n;i++)
{
int _min=MAX,note;
for(int j=1;j<=n;j++)
if(!vis[j]&&dis[j]<_min)
{
_min=dis[j];
note=j;
}
vis[note]=1;
int k=0;
for(int j=1;j<n;j++)
if(vis[j]&&_min==_map[note][j])
k++;
if(k>1)
{
printf("Not Unique!\n");
return;
}
sum+=_min;
for(int j=1;j<=n;j++)
if(!vis[j]&&dis[j]>_map[note][j])
dis[j]=_map[note][j];
}
printf("%d\n",sum);
}
void showmap()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
printf("%d ",_map[i][j]);
printf("\n");
}
}
int main()
{
//freopen("in","r",stdin);
int t;
cin>>t;
while(t--)
{
scanf("%d%d",&n,&m);
init();
for(int i=0;i<m;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
_map[a][b]=_map[b][a]=_map[a][b]>c?c:_map[a][b];
}
// showmap();
prim();
}
return 0;
}