数据结构实验之图论九：最小生成树（prime队列优化版）

顾客言

已于 2023-03-11 21:28:21 修改

阅读量67

点赞数 1

文章标签：数据结构图论算法 Powered by 金山文档

于 2023-03-01 15:41:18 首次发布

本文链接：https://blog.csdn.net/m0_73673533/article/details/129282883

版权

#include<bits/stdc++.h>
using namespace std;
const int N=2510;
int n,m;
typedef pair<int,int> PII;
vector<PII> g[N];
int dist[N];
bool vis[N];
int sum;
void prime()
{
    dist[1]=0;
    priority_queue<PII,vector<PII>,greater<PII>>q;
    q.push({0,1});
    while(q.size())
    {
        int u=q.top().second;
        q.pop();
        if(vis[u])continue;
        vis[u]=true;
        for(auto [v,w]:g[u])
        {
            if(!vis[v]&&dist[v]>w)
            {
                dist[v]=w;
                q.push({dist[v],v});
            }
        }
        sum+=dist[u];
    }
}
int main()
{
    while(cin>>n>>m)
    {
        sum=0;
    for(int i=1;i<=n;i++)
    g[i].clear();
    memset(dist, 0x3f, sizeof(dist)); 
    memset(vis, false, sizeof(vis));
    for(int i=0;i<m;i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        g[u].push_back({v,w});
        g[v].push_back({u,w});
    }
    prime();
    cout<<sum<<endl;
}
    return 0;
}