【图论】最小生成树

图结构练习——最小生成树

Time Limit: 1000MS Memory limit: 65536K

题目描述

 有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。

 

输入

 输入包含多组数据，格式如下。

第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n<=100)

剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。

输出

 每组输出占一行，仅输出最小花费。

示例输入

3 2
1 2 1
1 3 1
1 0

示例输出

2
0

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define MAXN 10000
#define inf 1<<28
int n, m;
int mapp[MAXN][MAXN];
int vis[MAXN], dis[MAXN];

void init()
{
    int i, j;
    for(i=1; i<=n; i++)
    {
        for(j=1; j<=n; j++)
        {
            mapp[i][j] = inf;
        }
        if(i==j)
            mapp[i][j] = 0;
    }
}
void prim()
{
    int i, j, pos, ans=0, min;
    for(i=2; i<=n; i++)
        dis[i] = mapp[1][i];
    vis[1] = 1;
    for(i=1; i<n; i++)
    {
        min = inf;
        for(j=2; j<=n; j++)
        {
            if(!vis[j] && dis[j]<min)
            {
                pos = j;
                min = dis[j];
            }
        }
        vis[pos] = 1;
        ans += min;

        for(j=2; j<=n; j++)
        {
             if(!vis[j] && mapp[pos][j]<dis[j])
                 dis[j] = mapp[pos][j];
        }
    }
    printf("%d
", ans);
}

int main()
{
     int i, j;
     int u, v, c;
     while(~scanf("%d%d", &n, &m))
     {
         init();
         memset(vis, 0, sizeof(vis));
         memset(dis, 0, sizeof(dis));
         for(i=0; i<m; i++)
         {
             scanf("%d%d%d", &u, &v, &c);
             if(mapp[u][v]>c)//判重边
             {
                 mapp[u][v] = c;
                 mapp[v][u] = c;
             }
         }
         prim();
     }
}