poj 1258 Agri-Net（最小生成树，普里姆算法）

小记：熟练了下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;
}