洛谷 P3366 【模板】最小生成树 prim算法思路 我自己的实现

网上有很多prim算法  用邻接矩阵 加什么lowcost数组 我觉得不靠谱 毕竟邻接矩阵本身就不是存图的好方法

所以自己写了一个邻接表（边信息表）版本的  注意我还是用了优先队列  每次新加入一个点  立即从这个点出发去查那些没有被选择的边与对面的点

优先队列来帮助排序 保证最顶上的一定是最小边

 #include<cstdio>
 #include<iostream>
 #include<cstdlib>
 #include<cstring>
 #include<queue>
 #include<algorithm>
 using namespace std;
 const int maxn=;
 const int maxm=;

 int tb,from[maxm],to[maxm],len[maxm],next[maxm],first[maxn],n,m,ans;
 bool is_select[maxn];//表示某个点i是否被选中
 int cnt;//表示剩余的点数量
 //建图
 void addedge(int a,int b,int c){
     tb++;
     from[tb]=a;to[tb]=b;len[tb]=c;
     next[tb]=first[a];
     first[a]=tb;
 }

 struct u{
     int a,c;
 };

 struct cmp{
     bool operator()(u a,u b){
         return a.c>b.c;
     }
 };

 priority_queue<u,vector<u>,cmp> hep;//优先队列

 void geted(int v)  //从v号点出发  寻找没有访问过的点的边  加入队列
 {
     is_select[v]=;
     int p=first[v];
     while(p!=)
     {
         if(is_select[to[p]]==)
         {
             u neo={to[p],len[p]};
             hep.push(neo);
         }
         p=next[p];
     }
 }

 int main()
 {
     scanf("%d %d",&n,&m);

     for(int i=,a,b,c;i<=m;i++)
     {
         scanf("%d%d%d",&a,&b,&c);
         addedge(a,b,c);
         addedge(b,a,c);
     }
     cnt=n-;
     geted();
     while(cnt> && !hep.empty()){
         u hd=hep.top();
         while(is_select[hd.a]){
             hep.pop();
             hd=hep.top();
         }
         hep.pop();
         ans+=hd.c;
         geted(hd.a);
         cnt--;
     }

     printf("%d",ans);
     return ;
 }