2022-04-16老张修路360笔试-prim构建最小生成树

原创已于 2022-04-20 12:09:30 修改 · 4.2k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #贪心算法

于 2022-04-16 17:37:52 首次发布

数据结构与算法同时被 2 个专栏收录

24 篇文章

订阅专栏

C++

21 篇文章

订阅专栏

该篇博客详细介绍了Prim算法用于构建最小生成树的过程，并针对代码中重复判断的问题进行了优化。作者不仅更新了代码以正确输出加入边的顺序，还提供了关键步骤和改进后的算法实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

prim算法构建最小生成树，返回最小权重和

using namespace std;

    int main() {
        //通用图数据存储
        int n, m;
        cin >> n;
        cin >> m;
        int u, v;//入边，出边
        vector<vector<int>> edge(n, vector<int>(n));
        for (int i = 0; i < m; ++i) {
            cin >> u >> v;
            cin >> edge[u][v];//这是邻接矩阵
            edge[v][u] = edge[u][v];
        }
        unordered_map<int,int> mp;//存已经加入的边
        vector<int> vt;
        //prim开始
        mp[0] = 1;
        vt.push_back(0);
        int ans = 0;
        int minLen;
        int node;
        for (int i = 0; i < n - 1; ++i) {
            minLen = INT_MAX;
            node = -1;
            for (auto e1:vt) {//for e1 in vt,离已选点最近的点加进来
                for (int k = 0; k < n; ++k) {
                    if (mp[k]) continue;
                    else if (edge[e1][k] == 0) continue;//无边
                    else {
                        if (edge[e1][k] < minLen) {
                            node = k;
                            minLen = edge[e1][k];
                        }
                    }
                }
            }
            ans += minLen;
            mp[node] = 1;
            vt.push_back(node);
        }
        for (auto num : vt) {
            cout << num << endl;
        }
        cout << ans << endl;

    }