kruskal,prim算法

生成下图的最小生成树
在这里插入图片描述
prim循环找到这一个集合外的最小边的点,加入该集合。
kruskal算法就是排序选择最小的,且不能构成环。排序没有自己实现调用了sort()函数,可能有的编译器运行不了。

#include <iostream>
#include "stdio.h"
#include <algorithm>
#include "stdlib.h"
#define N 5
#define inf 1000000
#define MAXE 100
using namespace std;
 struct edges
{
    int bv,tv,w; //边集类型,存储一条边的起始顶点bv、终止顶点tv和权w
};
edges ee[1000];
int pai(edges a,edges b)
{
    return a.w<b.w;
}
int seeks(int set[],int v);
void kruskal(edges ge[],int n,int e);
void prim(int v,int a[N][N]);
int main()
{

   // freopen("1.txt","r",stdin);
     printf("克鲁斯卡尔\n");
    int n,e; //n是图的结点数,e是图的结点数
       n=5,e=8;
    ee[0].bv=2; ee[0].tv=3; ee[0].w=1;
    ee[1].bv=4; ee[1].tv=5; ee[1].w=2;
    ee[2].bv=1; ee[2].tv=4; ee[2].w=3;
    ee[3].bv=3; ee[3].tv=5; ee[3].w=3;
    ee[4].bv=1; ee[4].tv=2; ee[4].w=4;
    ee[5].bv=1; ee[5].tv=3; ee[5].w=4;
    ee[6].bv=2; ee[6].tv=5; ee[6].w=5;
    ee[7].bv=4; ee[7].tv=3; ee[7].w=9;
    /*scanf("%d",&n);
    scanf("%d",&e);
    for(int i=0; i<e; i++)
    {
        scanf("%d",&ee[i].bv);
        scanf("%d",&ee[i].tv);
        scanf("%d",&ee[i].w);
    }*/


    sort(ee,ee+e,pai);//ee表示的边集图是按权值从小到大排列的
    kruskal(ee,n,e);
    int i,j,a[5][5];
     printf("普里姆\n");
    for(i=0; i<5; i++)
        for(j=0; j<5; j++)
            a[i][j]=inf;
    a[0][1]=4;  a[1][0]=4;
    a[0][3]=3;  a[3][0]=3;
    a[0][2]=4;  a[2][0]=4;
    a[2][3]=9;  a[3][2]=9;
    a[1][2]=1;  a[2][1]=1;
    a[3][4]=2;  a[4][3]=2;
    a[2][4]=3;  a[4][2]=3;
    a[1][4]=5;  a[4][1]=5;
    prim(0,a);
    return 0;
}
void prim(int v,int a[N][N])
{
    int lowcost[N],closest[N];//lowcost 为顶点间的最小距离,closest为最小距离上的邻接顶点
    int i,j,k,min1;
    for(i=0; i<N; i++)//lowcost,closest初始化
    {
        lowcost[i]=a[v][i];
        closest[i]=v;
    }
    lowcost[0]=0;
    closest[0]=-1;
    //寻找U 和 V 之间连接的最短距离的邻接点
    for(i=0; i<N; i++)
    {
        min1=inf;
        k=-1;
        for(j=0;j<N;j++)
        {
            if(closest[j]!=-1&&lowcost[j]<min1)
            {
                min1=lowcost[j];
                k=j;
            }
        }
        if(k==-1) break;
        cout<<"("<<closest[k]+1<<", "<<k+1<<") "<<lowcost[k]<<endl;
        closest[k]=-1;
        for(j=0; j<N; j++)//输出新的邻接顶点,并修改lowcost值
        {
            if(closest[j]!=-1&&a[k][j]<lowcost[j] )
            {

                lowcost[j]=a[k][j];  //修改lowcost值
                closest[j]=k;//修改邻接顶点
            }
        }
    }
}

//并查集
int seeks(int set[],int v)
{
    int i=v;
    while(set[i]>0) i=set[i];
    return i;
}

void kruskal(edges ge[],int n,int e)
{
    int set[MAXE],v1,v2,i,j;
    for(i=1; i<=n; i++)
        set[i]=0;
    i=0;
    j=1;
    while(i<e&&j<n)//按边权递增顺序,逐边检查该边是否应加入到生成树中
    {
        v1=seeks(set,ge[i].bv); //确定顶点v所在的连通集
        v2=seeks(set,ge[i].tv);
       // cout<<ge[i].bv<<" :"<<v1<<", "<<ge[i].tv<<" :"<<v2<<endl;
        if(v1!=v2) //当v1,v2不在同一顶点集合,确定该边应当选入生成树
        {
            cout<<"("<<ge[i].bv<<", "<<ge[i].tv<<") "<<ge[i].w<<endl;
            set[v1]=v2;
           // printf("set[%d]=%d\n",v1,v2);
            j++;
        }
        i++;
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值