利用Prim算法解决最小生成树

最小生成树

定义：图G的生成子图T，如果满足：

（1）是树（去掉任意边，图不联通）

（2）边的权值之和最小

则T叫做最小生成树

求解策略：贪心算法

约束：1.只是图里面的边 2.正好|V|-1条边 3.没有圈

这里，有两种算法

首先介绍Prim-小树长大算法

每次从已经建立好的树外面，找到一个距离现在的树最近的点，然后把它联上树，问题解决。

实现的方式，类似Dijkstra算法的搜索可以将算法的时间复杂度降到O(V*V)的数量级

利用dists[]数组，找一个最近的点，每次更新距离树的距离

例子：

08-图7 公路村村通 (30分)

现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:

输入数据包括城镇数目正整数N（≤1000）和候选道路数目M（≤3N）；随后的M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见，城镇从1到N编号。

输出格式:

输出村村通需要的最低成本。如果输入数据不足以保证畅通，则输出−1，表示需要建设更多公路。

输入样例:

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例:

12

这题就是求最小生成树的权重，在加边时，进行累加操作就可以了。

我一开始是盲搜最近的边，没有进行优化，结果时间复杂度到了O(v^3),后来发现这不是Prim正统思想。

后来的代码见下：

#include <iostream>
#define inf 9999999

using namespace std;
struct way
{
    int price;
    /* data */
} graph[5300][5300];
int N,M;
int intree[5300];
int dists[5300];
int cost;
void ini()
{
    cin>>N>>M;
    int i,x,y,c;
    for(i=1;i<=M;i++)
    {
        cin>>x>>y>>c;
        graph[x][y].price=c;
        graph[y][x].price=c;
    }
}
int near_by_intree(int n)
{
    int i;
    for(i=1;i<=N;i++)
    {
        if(graph[i][n].price!=0&&intree[i]==0){
            return 0;
        }
    }
    return 1;
}
int find_dmin()
{
    int i;
    int md=inf-8,loca=-1;
    for(i=1;i<=N;i++)
    {
        if(intree[i]==0&&dists[i]<md)
        {
            md=dists[i];
            loca=i;
        }
    }
    if(loca==-1)
    {
        return -1;
    }
    else
    {
        cost+=md;
        return loca;
    }
}
int miner(int a,int b)
{
    if(a>b){
        return b;
    }
    else
    {
        return a;
    }

}

int prim(int start)
{
    intree[start]=0;
    int p=start;
    int i,j,k;
    for(i=0;i<N+1;i++)
    {
        dists[i]=inf;
    }
    dists[start]=0;
    while (1)
    {
        p=find_dmin();
        if(p==-1)
        {
            break;
        }
        intree[p]=1;
        for(i=1;i<=N;i++)
        {
            if(graph[i][p].price!=0&&intree[i]==0)
            {
                dists[i]=miner(graph[i][p].price,dists[i]);
            }
        }

    }
    for(i=1;i<=N;i++)
    {
        if(intree[i]==0)
        {
            cout<<"-1";
            return 0;
        }
    }
    cout<<cost;
}
int main()
{
    ini();
    int i,j;
    int pri[20000];
    prim(1);


}