POJ 1258 最小生成树

23333333333

完全是道水题。因为是偶自己读懂自己做出来的。。T_T、prim的模板题水过。

DESCRIPTION：
John竞选的时候许诺会给村子连网。现在给你任意两个村子之间的距离。让你求任意两个村庄是连通的所需要的网线。就是求最小生成树的权值。

附代码：

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<algorithm>
#define inf 0x1f1f1f1f
using namespace std;

int map[210][210];
int n;

int prim()
{
    int low[210], vis[210];
    int min, res = 0, p;
    memset(vis, 0, sizeof(vis));
    vis[0] = 1;
    for (int i=1; i<n; ++i)
    {
        low[i] = map[0][i];
    }
    for (int i=1; i<n; ++i)  // 循环n-1次 找最小生成树的n-1个点
    {
        min = inf;
        p = -1;
        for (int j=0; j<n; ++j)
        {
            if (vis[j] == 0 && min > low[j])
            {
                min = low[j];
                p = j;
            }
        }
        if (res == inf)
        return -1;
        res += min;
        vis[p] = 1;
        for (int j=0; j<n; ++j)
        {
            if (vis[j] == 0 && low[j] > map[p][j])
            {
                low[j] = map[p][j];
            }
        }
    }
    return res;
}

int main()
{
    while(cin >> n)
    {
        memset(map, 0, sizeof(map));
        for (int i=0; i<n; ++i)
        {
            for (int j=0; j<n; ++j)
            cin >> map[i][j];
        }
        int ans = prim();
        cout << ans << endl;
    }
    return 0;
}