最小生成树

kruskal算法(好像并查集)

#include<bits/stdc++.h>
using namespace std;
const int maxn = 114514;
int maxx,ans;
int fa[maxn],n,m,k=0;
int find(int x)
{
    if(fa[x]!=x)
    fa[x]=find(fa[x]);
    return fa[x];
}
struct node
{
    int x,y,w;
}a[maxn];
bool cmp(node x,node y)
{
    return x.w<y.w;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fa[i]=i;
    }    
    for(int i=1;i<=m;i++)
    {
        cin>>a[i].x>>a[i].y>>a[i].w;
    }
    sort(a+1,a+m+1,cmp);
    for(int i=1;i<=m;i++)
    {
        int ax=find(a[i].x),ay=find(a[i].y);
        if(ax!=ay)
        {
            fa[ax]=ay;
            k++;
            ans=a[i].w;
            if(k==n-1)
            {
                break;
            }
        }
    }
    cout<<n-1<<" "<<ans;
    return 0;
}

prim算法

#include<bits/stdc++.h>
#define min min1
using namespace std;
int n,m,maxx=9999999,min=99999999,c[2001],k;
bool b[2001];
int f[2001][2001];
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>f[0][i];
        f[i][0]=f[0][i];
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cin>>f[i][j];
        }
    }
    for(int i=1;i<=n;i++)
    {
        f[i][i]=maxx;
    }
    int cnt=0;
    b[0]=1;
    c[0]=0;
    for(int i=1;i<=n;i++)
    {
        c[i]=f[0][i];
    }
    for(int i=1;i<=n;i++)
    {
        min=maxx;
        k=0;
        for(int j=1;j<=n;j++)
        {
            if((!b[j]) and (c[j]<min))    
            {
                min=c[j];
                k=j;
            }
        } 
        cnt+=c[k];
        b[k]=1;
        for(int j=1;j<=n;j++)
        {
            if(f[k][j]<c[j])
            c[j]=f[k][j];
        }
    }
    cout<<cnt<<endl;
    return 0;
}

posted @ 2024-03-12 20:27  类人群星闪耀时  阅读(6)  评论(0编辑  收藏  举报