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;
}