其他分享
首页 > 其他分享> > 生成树之普利姆生成树(模板)

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

作者:互联网

不同于克鲁斯卡尔生成树,每次寻找权值最小的边进行连接,普利姆生成树的规则是,选取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;
}

标签:普利,int,生成,1000,节点,模板,dis
来源: https://blog.csdn.net/m0_67394360/article/details/123581387