最小生成树的算法

今天写了一个最小生成树的算法，也是为了准备蓝桥杯。题解再说，先专门讨论算法。

最小生成树的算法原理我一直都知道，就是代码一直不会写。简单说一下原理好了。一个无向无环图，每条边上有相应的权值，找到权值的和最小的生成树就是最小生成树。首先将所有的点分为两个集合U，V。其中U是已经找到对应边的点。每次寻找U到V的最短边，并且把这条边在V中的端点加入U。循环下去直到所有的点都被找到。

灵感来源于http://blog.chinaunix.net/uid-25324849-id-2182922.html

下面是代码，注释很清楚了（不连通的点距离是无穷大，自己到自己也是无穷大）

#include<iostream>
#include<stack>
#include<stdio.h>
#include<string.h>
using namespace std;
const int in = 1000;
int main()
{
    int map[5][5] = {{in,30,in,in,in},{30,in,40,40,70},{in,40,in,60,62},{in,40,60,in,60},{in,70,62,60,in}};
    int cost[5];//当前点集合内的点到其他点的最短距离
    int point[5];//在集合内的边（i，point[i]）
    bool visit[5];//该点在不在集合内
    int father[5];//好像没什么用
    int i;
    //先将第一个点加入集合内并且给数组们赋值
    for(i = 0;i < 5;i++)
    {
        cost[i] = map[0][i];
        point[i] = 0;
        visit[i] = false;
        father[i] = 0;
    }
    visit[0] = true;
    int k;
    //有n个顶点就有n-1条边，进行n-1次循环求出这些边
    for(k = 1;k < 5;k++)
    {
        int j = 0;
        int min = in;
        //找出不在集合内的与集合内的点距离最短的点
        for(i = 0;i < 5;i++)
        {
            
            if(!visit[i] && min > cost[i]){min = cost[i];j = i;}
        }
        //将这个点加入集合
        visit[j] = true;
        father[j] = point[j];
        //更新数组们
        for(i = 0;i < 5;i++)
        {
            //对于还不在集合内点，如果与新加入的点的距离小于原来的距离就改变cost数组，
            //并且将该点的边的另一端改为新加入的点
            //i是不在集合里的点，j是刚才加入的点
            if(!visit[i] && cost[i]>map[j][i]){cost[i] = map[j][i];point[i] = j;}
        }
        cout<<j<<" "<<point[j]<<endl;//打印出这个新加入的边
    }
    int sum = 0;
    for(i = 1;i < 5;i++)
    {
        sum += map[i][point[i]];
    }
    cout<<sum;//最小生成树的权值之和
    return 0;
}