小记:熟练了下prim,其实是照着dijsktra来写的,就一个判断条件的不同而已
思路:prim算法
代码:
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
#include <string>
using namespace std;
#define mst(a,b) memset(a,b,sizeof(a))
#define REP(a,b,c) for(int a = b; a < c; ++a)
#define eps 10e-8
const int MAX_ = 155;
const int N = 100010;
const int INF = 0x7fffffff;
int n;
struct node{
int s, t;
};
int g[MAX_][MAX_];
int d[MAX_];
bool vis[MAX_];
int m, cnt;
int prim(int start)
{
REP(i, 0, n){
d[i] = INF;
vis[i] = 0;
}
int sum = 0;
d[start] = 0;
REP(i, 0, n){
int mmin = INF, k;
REP(j, 0, n){
if(!vis[j] && d[j] < mmin){
mmin = d[j];
k = j;
}
}
vis[k] = 1;
sum += mmin;
REP(j, 0, n){
if(!vis[j] && d[j] > g[k][j]){
d[j] = g[k][j];
}
}
}
return sum;
}
int main(){
int T, ss, tt;
while(~scanf("%d", &n)){
REP(i, 0, n)REP(j, 0, n){
scanf("%d", &g[i][j]);
}
int ans = prim(0);
printf("%d\n", ans);
}
return 0;
}