prim算法

最小生成树板子题：POJ 1287

---

最小生成树的prim算法一直没学，而且好像最短路也会用到这种思路，于是今天学了一下，找了很多博客，算法图解都挺好的，但是代码实现就很复杂，最后找了篇模板题题解博客看了一遍才懂代码实现。

prim算法其实思路很简单，就是不断的找当前距离已生成的树最小的结点加入树，直到所有结点都入树就完成了。

ps：如果算法思路都不清楚的话，可以看看这篇博客的图解：传送门（代码就不建议看了，个人感觉写的不太友好）

知道了思路，接下来就看上面的模板题，然后仔细、耐心的模拟一遍下面的代码应该就没问题了。

//https://blog.csdn.net/hesorchen
#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
#include <queue>
#include <stack>
#include <list>
#include <map>
#include <set>
using namespace std;
#define ll long long
#define endl "\n"
#define INF 0x3f3f3f3f
#define mod 1000000007
#define MAX 55

int n, m;
int mp[MAX][MAX]; //邻接矩阵
int d[MAX];       //表示每个图外的结点距离已生成树的距离
int t[MAX];       //记录生成树内外的结点
int prim()
{
    int ans = 0;
    d[1] = 0; //假设结点1在生成树内，其他都在树外
    int v;
    while (1)
    {
        v = -1;
        for (int i = 1; i <= n; i++)
        {
            if (!t[i] && (v == -1 || d[i] < d[v])) //寻找当前离生成树最近的结点v
                v = i;
        }
        if (v == -1) //如果所有结点都已经在生成树内，说明最小生成树已经生成
            break;
        t[v] = 1; //把结点v加入结点
        ans += d[v];
        for (int i = 1; i <= n; i++) //把结点v加入生成树之后，要更新与v相连的结点 到已生成树的距离！！！
            d[i] = min(d[i], mp[i][v]);
    }
    return ans;
}
int main()
{
    while (cin >> n && n)
    {
        cin >> m;
        for (int i = 1; i <= n; i++) //初始化邻接矩阵
        {
            for (int j = 1; j <= n; j++)
                if (i == j)
                    mp[i][j] = 0;
                else
                    mp[i][j] = INF;
        }
        while (m--)
        {
            int x, y, z;
            cin >> x >> y >> z; //输入更新邻接矩阵
            if (mp[x][y] > z)
                mp[x][y] = mp[y][x] = z;
        }
        for (int i = 1; i <= n; i++)
            d[i] = INF;
        fill(t, t + MAX, 0);
        cout << prim() << endl;
    }
    return 0;
}

今天效率好低，只看了这么一种算法。。。（太浮躁了！！）