POJ1258——最小生成樹

題目描述：

有n個農場，兩兩之間有路，求連接所有農場最小的花費。

大致思路：

最小生成樹的模板題。Prim，克魯斯卡爾……等等，能搞就行！

代碼：

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int inf = 0x3f3f3f3f;
const int maxn = 100 + 10;
int g[maxn][maxn];
int n;

int prim() {
    int dist[maxn];
    bool vis[maxn];
    memset(vis,0,sizeof(vis));
    memset(dist,inf,sizeof(dist));
    dist[1] = 0;
    while (1) {
        int l = 0;
        for (int i = 1; i <= n; i++) {
            if (!vis[i] && dist[i] < dist[l]) l = i;
        }
        if (!l) break;
        for (int j = 1; j <= n; j++) {
            if (j == l) continue;
            if (!vis[j]) dist[j] = min(dist[j],g[l][j]);
        }
        vis[l] = true;
    }
    int ans = 0;
    for (int i = 1; i <= n; i++) ans += dist[i];
    return ans;
}

int main() {
    while (cin>>n) {
        memset(g,0,sizeof(g));
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= n; j++) {
                scanf("%d",&g[i][j]);
            }
        }
        cout<<prim()<<endl;
    }
}