Networking POJ-1287【最小生成树】

题目链接：POJ-1287

[kuangbin带你飞]专题六最小生成树

题目描述：

您被分配设计广泛区域中某些点之间的网络连接。您将获得该区域中的一组点，以及可连接成对点的电缆的一组可能路线。对于两点之间的每条可能路线，您将获得连接该路线上的点所需的电缆长度。请注意，在两个给定点之间可能存在许多可能的路径。假设给定的可能路线（直接或间接）连接该区域中的每两个点。
您的任务是为该区域设计网络，以便在每两个点之间存在连接（直接或间接）（即，所有点都是互连的，但不一定是通过直接电缆），并且总长度为用过的电缆很少。

思路

Prim算法：
Prim算法就是定义mat用来存图，dist数组用来存每个点到最小生成树的最短距离，vis数组标记此点是否已经在最小生成树中了（防止出现环）。
实际操作就是先随便把一个点加入到生成树里（一般是第一个点），然后寻找与该点距离最短的点并加入，然后在寻找与1,2点相连最短的点，并加入到生成树中，就这样不断的加下去，直到加入的点数与原图一致终止；
可以自己手动模仿一下过程，其实还是好理解的！
##易错点
1.每次加入前确保加入的权值比之前加入的权值要小，否则不加入；
如 第一次 给出 1-2边的权值为3， 然后某次加边操作说 1-2边权值为10， 这时候我们保留第一次的权值（保留权值较小的那次）
2.题意说是边可能无限，那么大概不能用Kruskal算法来做；Prim算法用来存顶点，而Kruskal算法用来存边，当边很多或者不确定是，我们可能无法用Kruskal算法来做（当然，现在2018年10月9日16:39:24可能还是不能做的，以后说不定也可以做）

代码:

Prim算法：

#include <iostream>
#include <string.h>
using namespace std;

#define MAX 5050
#define INF 0x3f3f3f3f

int mat[MAX][MAX], vis[MAX], dist[MAX];

void init(int n)
{
    for(int i = 1; i<=n; i++)
    {
        for(int j = 1; j<=n; j++)
        {
            mat[i][j] = INF;
        }
        dist[i] = INF;
    }
}

void Union(int x, int y, int val)
{
    if(val < mat[x][y])
    {
        mat[x][y] = mat[y][x] = val;
    }
}

int Prim(int n)
{
    int sum = 0;
    dist[1] = 0;
    for(int i = 1; i<=n; i++)
    {
        int min_dist = INF, min_vertex;
        for(int j = 1; j<=n; j++)
        {
            if(!vis[j] && dist[j] < min_dist)
            {
                min_dist = dist[j];
                min_vertex = j;
            }
        }
        vis[min_vertex] = 1;
        sum += min_dist;
        for(int j = 1; j<=n; j++)
        {
            if(!vis[j] && mat[min_vertex][j] < dist[j])
            {
                dist[j] = mat[min_vertex][j];
            }
        }
    }
    return sum;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n, m;
    while(cin >> n, n != 0)
    {
        cin >> m;
        init(n);
        memset(vis, 0, sizeof(vis));
        for(int i = 0; i<m; i++)
        {
            int x, y, val;
            cin >> x >> y >> val;
            Union(x, y, val);
        }
        cout << Prim(n) << endl;
    }
    return 0;
}