最小生成树MST-Prime算法(堆优化)

//给定一个 n 个点 m 条边的无向图，图中可能存在重边和自环，边权可能为负数。
//时间复杂度O(mlogn) 适用于稠密图
#include<bits/stdc++.h>
#define N 1000000
using namespace std;
int n,m,cnt,vis[N],dist[N],head[N],ans;
typedef pair<int,int>PII;
struct edge
{
    int to,next,weight;
}e[N];
void add(int u,int v,int w)
{
    e[++cnt].to=v;
    e[cnt].weight=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}
void Prim()
{
    memset(dist,0x3f,sizeof(dist));
    priority_queue<PII,vector<PII>,greater<PII> >heap;
    dist[1]=0;
    heap.push({0,1});
    while(!heap.empty())
    {
        PII t=heap.top();
        heap.pop();
        vis[t.second]=1;
        for(int i=head[t.second];i;i=e[i].next)
        {
            if(!vis[e[i].to]&&e[i].weight<dist[e[i].to])
            {
                dist[e[i].to]=e[i].weight;
                heap.push({e[i].weight,e[i].to});
            }
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
        add(v,u,w);
    }
    Prim();
    for(int i=1;i<=n;i++)
    {
        if(dist[i]==0x3f3f3f3f)
        {
            cout<<"impossible";
            exit(0);
        }
    }
    for(int i=2;i<=n;i++) ans+=dist[i];
    cout<<ans;
    return 0;
}