首页 > 代码库 > 最小生成树prim
最小生成树prim
矩阵表示。
#include<iostream> #include<cstring> #include<cstdio> #include<algorithm> #define INF 0x3f3f3f3f using namespace std; int a[1005][1005],vis[1005],cost[1005],n,m; int prim() { int ans = 0; memset(vis,0,sizeof(vis)); vis[1] = 1; for(int i = 2;i <= n;i++) cost[i] = a[1][i]; for(int i = 2;i <= n;i++) { int minn = INF; int k = -1; for(int j = 1;j <= n;j++) { if(!vis[j] && minn > cost[j]) { minn = cost[j]; k = j; } } if(minn == INF) return -1; ans += minn; vis[k] = 1; for(int j = 1;j <= n;j++) { if(!vis[j] && cost[j] > a[k][j]) cost[j] = a[k][j]; } } return ans; } int main() { memset(a,0x3f,sizeof(a)); scanf("%d%d",&n,&m); for(int i = 1;i <= m;i++) { int u,v,w; scanf("%d%d%d",&u,&v,&w); a[u][v] = a[v][u] = w; } printf("%d\n",prim()); return 0; } //PTA公路村村通 //https://pta.patest.cn/pta/test/1342/exam/4/question/24938
最小生成树prim
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。