图结构练习——最小生成树 Prim算法

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

Time Limit: 1000ms   Memory limit: 65536K  有疑问？点这里^_^

题目描述

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

 

输入

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

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

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

 

输出

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

示例输入

3 2
1 2 1
1 3 1
1 0

示例输出

2
0

提示

 

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#define INF 0x3f3f3f3f

using namespace std;
int n,m;
bool vis[120];
int Dis[120];
int Map[120][120];
int prim()
{
    memset(vis,false,sizeof(vis));
    for(int i=1; i<=n; i++)
    {
        Dis[i]=Map[1][i];
    }
    Dis[1]=0;
    vis[1]=true;
    int sum=0;
    for(int i=1;i<n;i++)
    {
        int ans=INF;
        int flag;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&Dis[j]<ans)
            {
                ans=Dis[j];
                flag=j;
            }
        }
        vis[flag]=true;
        sum+=ans;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&Dis[j]>Map[flag][j])
            {
                Dis[j]=Map[flag][j];
            }
        }
    }
    return sum;
}
int main()
{
    int u,v,w;
    while(~scanf("%d %d",&n,&m))
    {
        memset(Map,INF,sizeof(Map));
        for(int i=0; i<m; i++)
        {
            scanf("%d %d %d",&u,&v,&w);
            if(Map[u][v]>w)
            {
                Map[u][v]=Map[v][u]=w;
            }
        }
        cout<<prim()<<endl;
    }

    return 0;
}
//1).输入：一个加权连通图，其中顶点集合为V，边集合为E；
//2).初始化：Vnew = {x}，其中x为集合V中的任一节点（起始点），Enew = {},为空；
//3).重复下列操作，直到Vnew = V：
//a.在集合E中选取权值最小的边<u, v>，其中u为集合Vnew中的元素，而v不在Vnew集合当中，
//并且v∈V（如果存在有多条满足前述条件即具有相同权值的边，则可任意选取其中之一）；
//b.将v加入集合Vnew中，将<u, v>边加入集合Enew中；
//4).输出：使用集合Vnew和Enew来描述所得到的最小生成树。