#include<bits/stdc++.h> using namespace std; #define INF 0x3f3f3f3f const int maxn = 117; int m[maxn][maxn]; int vis[maxn], low[maxn]; /* 對于這道題目來將,m就是臨接矩陣,vis是訪問標記數組,low是最短距離數組 */ int n; int prim() {vis[1] = 1;int sum = 0;int pos, minn;pos = 1;for(int i = 1; i <= n; i++){low[i] = m[pos][i];}/*先把第一個點放到樹里,然后找到剩下的點到這個點的距離*/for(int i = 1; i < n; i++)//循環遍歷 n-1 次數,把點全部加入!{minn = INF;for(int j = 1; j <= n; j++){if(!vis[j] && minn > low[j]) //沒有進樹的節點,并且這個節點到樹里面 點距離最近,拉進來{minn = low[j];pos = j;}}sum += minn;vis[pos] = 1;for(int j = 1; j <= n; j++){if(!vis[j] && low[j] > m[pos][j])//用新加入的點,更新low值{low[j] = m[pos][j];}}}return sum; } void init() {memset(vis,0,sizeof(vis));memset(low,0,sizeof(low));for(int i = 1; i <= n ;i++ )for(int j = 1; j <= n; j++)m[i][j] = INF; } void in_map() {printf("輸入鄰接矩陣階:\n");scanf("%d",&n);printf("輸入鄰接矩陣,無窮用 -1代表!\n");int t;for(int i=1;i<=n;i++)for(int j=1;j<=n;j++){scanf("%d",&t);m[i][j] = (t==-1?INF:t);} } int main() {init();in_map();printf("%d",prim()); }
轉載于:https://www.cnblogs.com/zpf1/p/9070776.html