生成树之普利姆生成树(模板)

24 篇文章 0 订阅

不同于克鲁斯卡尔生成树,每次寻找权值最小的边进行连接,普利姆生成树的规则是,选取1节点为树根,每次寻找最靠近生成树的节点加入生成树,并修改树外节点到树的距离。

普利姆适应于点少的情况,当题目给出点坐标而非边权值时,往往利用克鲁斯卡尔会陷入对大量边的计算,这时普利姆的优势得以体现。

细节如下

一,初始化dis数组为inf

二,dis[i]数组赋初值时,赋成w[1][i],表示到根节点距离

三,标记数组事先标记1节点

四,每次找寻树外节点(未被标记),找到之后别忘了标记,并且记录答案

五,借助这一节点去更新树外节点dis值

# include<iostream>
# include<math.h>
# define inf 0x7f7f7f7f
using namespace std;
typedef long long int ll;

int w[1000][1000];
bool book[1000];
int dis[1000];
int n;
int main ()
{

    fill(dis,dis+1000,99999);

    cin>>n;

    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cin>>w[i][j];
        }
    }


    for(int i=1;i<=n;i++)
    {
        dis[i]=w[1][i];
    }

    dis[1]=0;
    book[1]=1;
    int now,min1,sum=0;

    for(int i=1;i<n;i++)   //找寻n-1个点
    {
        min1=inf;

        for(int j=1;j<=n;j++)
        {
            if(book[j]==0)
            {
                if(dis[j]<min1)
                {
                    min1=dis[j];
                    now=j;
                }
            }
        }
        sum+=min1;

        book[now]=1;


        for(int j=1;j<=n;j++)
        {
            if(book[j]==0)
            {
                if(dis[j]>w[now][j])
                {
                    dis[j]=w[now][j];
                }
            }
        }

    }

    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值