最小生成树模板

最新推荐文章于 2025-03-04 15:20:31 发布

转载最新推荐文章于 2025-03-04 15:20:31 发布 · 51 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/pach/p/5738306.html

prim：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int inf=1000000;
const int maxn=105;
int mat[maxn][maxn],dis[maxn];
bool vis[maxn];
int n,ans;

bool prim()
{
    memset(vis,0,sizeof(vis));
    for(int i=0;i<n;i++)
        dis[i]=inf;
    ans=0;dis[0]=0;
    for(int i=0;i<n;i++)
    {
        int temp=inf,k=0;
        for(int j=0;j<n;j++)
        {
            if(!vis[j] && dis[j]<temp )
            {
                temp=dis[j];
                k=j;
            }
        }
        if(temp==inf)
            return false;
        vis[k]=true;
        ans+=temp;
        for(int j=0;j<n;j++)
        {
            if(!vis[j] && dis[j]>mat[k][j])
                dis[j]=mat[k][j];
        }
    }
    return true;
}