数据结构实验之图论九：最小生成树

数据结构实验之图论九：最小生成树

Problem Description
有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。
Input
输入包含多组数据，格式如下。
第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)
剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。
Output
每组输出占一行，仅输出最小花费。
Sample Input
3 2
1 2 1
1 3 1
1 0
Sample Output
2
0

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#define INF 0x3f3f3f3f
using namespace std;
int vis[1100],queue[1100],dis[1100],mon[1100],map[1100][1100];;
int n,m,t,s,d,head,tail;
void prime()
{
    int k,sum=0;
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)
    dis[i]=map[1][i];
    vis[1]=1;
    for(int i=1;i<n;i++)
    {
        int MIN=INF;
        for(int j=1;j<=n;j++)
        if(vis[j]==0&&dis[j]<MIN){MIN=dis[j];k=j;}
        vis[k]=1;sum+=MIN;
        for(int j=1;j<=n;j++)
            if(!vis[j])
            dis[j]=min(dis[j],map[k][j]);
    }
    printf("%d\n",sum);
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
        {
            if(i==j)map[i][j]=0;
            else map[i][j]=INF;

        }
        while(m--)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            if(w<map[u][v])map[u][v]=map[v][u]=w;
        }
        prime();
    }
    return 0;
}

#include <stdio.h>
#include <stdlib.h>
int f[11000];
struct node
{
    int a,b,c;
} aa[11000];
int cmp(const void *a,const void *b)
{
    return (*(struct node *)a).c-(*(struct node *)b).c;
}
int find(int x)
{
    if(x!=f[x])
    {
        f[x]=find(f[x]);
    }
    return f[x];
}
int join(int x,int y)
{
    int t1=find(x);
    int t2=find(y);
    if(t1!=t2)
    {
        f[t2]=t1;
        return 1;
    }
    return 0;
}
int main()
{
    int n,m,i,t,s;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(i=1; i<=n; i++)
        {
            f[i]=i;
        }
        for(i=0; i<m; i++)
        {
            scanf("%d%d%d",&aa[i].a,&aa[i].b,&aa[i].c);
        }
        qsort(aa,m,sizeof(aa[0]),cmp);
        t=0,s=0;
        for(i=0; i<m; i++)
        {
            if(join(aa[i].a,aa[i].b))
            {
                t++;
                s=s+aa[i].c;
            }
            if(t==n-1) break;
        }
        printf("%d\n",s);
    }
    return 0;
}