图论九——最小生成树（prim）

ゝ一纸荒年。 2022-06-05 08:27 159阅读 0赞

数据结构实验之图论九：最小生成树  
Time Limit: 1000MS Memory Limit: 65536KB  
Submit Statistic  
Problem Description

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

Input

输入包含多组数据，格式如下。  
第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)  
剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。

Output

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

3 2  
1 2 1  
1 3 1  
1 0  
Example Output

2  
0  
Hint

Author

赵利强

#include <iostream>
    #include <bits/stdc++.h>
    
    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 = 0x3f3f3f3f;
            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(cin>>n>>m)
        {
            memset(Map,0x3f3f3f3f,sizeof(Map));
            for(int i = 0; i < m; i++)
            {
                cin>>u>>v>>w;
                if(Map[u][v]>w)
                {
                    Map[u][v] = Map[v][u] = w;
                }
            }
            cout<<prim()<<endl;
        }
        return 0;
    }
    
    
    /***************************************************
    Result: Accepted
    Take time: 40ms
    Take Memory: 260KB
    Submit time: 2017-06-10 20:56:18
    ****************************************************/