#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()); }