ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

Prim算法学习

2022-02-09 22:02:51  阅读:166  来源: 互联网

标签:输出 Prim int 最小 生成 学习 算法 impossible prim


模板题

Prim算法求最小生成树

给定一个 n 个点 m 条边的无向图,图中可能存在重边和自环,边权可能为负数。

求最小生成树的树边权重之和,如果最小生成树不存在则输出 impossible。

给定一张边带权的无向图 G=(V,E),其中 V 表示图中点的集合,E 表示图中边的集合,n=|V|,m=|E|。

由 V 中的全部 n 个顶点和 E 中 n−1 条边构成的无向连通子图被称为 G 的一棵生成树,其中边的权值之和最小的生成树被称为无向图 G 的最小生成树。

输入格式
第一行包含两个整数 n 和 m。

接下来 m 行,每行包含三个整数 u,v,w,表示点 u 和点 v 之间存在一条权值为 w 的边。

输出格式
共一行,若存在最小生成树,则输出一个整数,表示最小生成树的树边权重之和,如果最小生成树不存在则输出 impossible。

数据范围
1≤n≤500,
1≤m≤105,
图中涉及边的边权的绝对值均不超过 10000。

输入样例:
4 5
1 2 1
1 3 2
1 4 3
2 3 2
3 4 4
输出样例:
6

代码

//稠密图-朴素prim-代码短O(n^2)
//堆优化prim-O(mlogn)-不常用
#include<bits/stdc++.h>
using namespace std;
const int N=505,INF=0x3f3f3f3f;
int g[N][N],n,m,x,y,z;
int d[N];bool st[N];
int prim()
{   
    memset(d,0x3f,sizeof d);int res=0;

    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
        {
            if(!st[j]&&(t==-1||d[j]<d[t]))
            t=j;
        }
        if(i&&d[t]==INF)return INF;
        if(i)   res+=d[t];  //需在循环更新之前累加!!!
        st[t]=true;
        
        
        for(int j=1;j<=n;j++)
        {
            d[j]=min(d[j],g[t][j]);
        }
      
    }
    return res;
}
int main()
{
  //ios::sync_with_stdio(false);cin.tie(0); cout.tie(0);
    memset(g,0x3f,sizeof g);
    cin>>n>>m;
    while(m--)
    {
        cin>>x>>y>>z;
       g[y][x]=g[x][y]=min(g[x][y],z);
    }
    int t=prim();
    if(t==INF)
    puts("impossible");
    else
    cout<<t<<endl;
    system("pause");
    return 0;
}

标签:输出,Prim,int,最小,生成,学习,算法,impossible,prim
来源: https://www.cnblogs.com/zp1125/p/15876950.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有