最小生成树

Prime算法
编写一个程序,计算给定加权图G=(V,E)的最小生成树的各边权值之和。
输入:第一行输入G的顶点数n,接下来n行输入表示G的n*n邻接矩阵A。A的元素 aij 代表顶点 i 到顶点 j 的边的权值,边不存在时为-1.
输出:输出G的最小生成树的各边权值总和。
题目  https://cn.vjudge.net/problem/Aizu-ALDS1_12_A

1.首先定义一个color数组,先将图中所有的结点都初始为“WHITE”(white代表的就是此结点没有被放入集合t中),并且定义一个 d[]数组,d[] 数组用于记录连接集合t内顶点与集合s内顶点的边中权值最小的边的权值。最后定义一个p[] 数组,这个数组用于记录集合 t 中顶点的父结点。
2.开始时先把u初始化为-1,d[0] = 0。通过遍历找出最小权值的边,并且连接这条边的两个顶点都没有放入到t集合中,用变量u记录这条边的末位置。
3.随后从u这个末位置出发寻找与其相连接的其他边的权值,此时应该边遍历边更新d[i]的值(永远选择到达那个顶点权值最小的那个),并且记录p[i]。
4.重复上述2,3步骤,如果2中遍历一遍之后u的值依然为-1,代表所有的点都已经放入集合t中了,则跳出循环。
5.最后累加每个点线连接线的权值(此时p[i]就可起到作用)。

#include <iostream>
using namespace std;
#define MAX 105
#define NIL (1 << 21)
#define WHITE 0
#define BLACK 1

int n;
int a[MAX][MAX];

//prime算法的实现
int prime()
{
    int color[MAX],d[MAX],p[MAX],min,u,sum;         //color[]记录顶点的访问状态 
    for(int i = 0;i < n;i++)                         // d[]记录权值最小边的权值 p[]记录选定节点的父亲节点 
    {
        color[i] = WHITE;
        d[i] = NIL;
    }
    d[0] = 0;                                        //初始的时候,与第一个结点连接的权值为0
    while(1)
    {
        min = NIL;
        u = -1;
        for(int i = 0;i < n;i++)
        {
            if(min > d[i] && color[i] != BLACK)    //超出权值最小的那条边,并且将已纳入的点除外
            {
                u = i;
                min = d[i];
            }
        }
        if(u == -1) break;                            //当所有点都为黑的时候跳出
        color[u] = BLACK;                             //将选中的点变黑

        for(int v = 0;v < n;v++)
        {
            if(a[u][v] != -1 && color[v] != BLACK)    //排除形成环
            {
                if(d[v] > a[u][v])
                {
                    d[v] = a[u][v];
                    p[v] = u;                         //标记连接点的起始位置
                }
            }
        }
    }
    sum = 0;
    for(int i = 1;i < n;i++) 
	    sum += a[p[i]][i];
    return sum;
}

int main()
{
    scanf("%d",&n);
    for(int i = 0;i < n;i++)
    {
        for(int j = 0;j < n;j++) 
		scanf("%d",&a[i][j]);
    }

    cout << prime() << endl;
    return 0;
}

Kruskal算法

1.给所有边从小到大排序

2.考察每条边(u,v)选择不在同一连通分量,不能形成回路的边

分析:

关键点在于“连通分量的查询与合并”——需要知道任意两个点是否在同一个连通分量中,还需要合并两个连通分量。

解决方法:

(1)     暴力求解法:

(2)     使用并查集(Union-Find Set)

//Kruskal算法
//假设第i条边的两个端点序号和权值分别保存在u[i],v[i]和w[i]中
//排序后第i小的边的序号保存在r[i]中
int cmp(const int i,const int j)             //间接排序函数
{
    return w[i]<w[j];
}
int find(int x)                             
{
    return p[x] == x ? x : p[x] = find(p[x]);   //如果 p[x] = x,说明x本身是树根,返回x;否则返回p[x]的父节点所在树的树根
}
int Kruskal()
{
    int ans = 0;
    for(int i = 0;i < n;i++)                   //初始化并查集
        p[i]=i;
    for(int i = 0;i < m;i++)                   //初始化边序号
        r[i]=i;
    sort(r ,r+m ,cmp);                        //对r[i]数组按w[i]从小到大进行排序
    for(int i = 0; i<m; i++)
    {
        int e = r[i];                         //e是取出边的下标 
	  int x = find(u[e]);                   //找到边的左节点所对应的连通子集编号
        int y = find(v[e]);                   //找到边的右节点所对应的连通子集编号
        if(x !=y)                             //如果在不同连通分量合并
        {
            ans += w[e];
            p[x] = y;
        }
    }
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值