AcWing 1140. 最短网络最小生成树板子题

Posted 2021-11-01 幽殇默

https://www.acwing.com/problem/content/1142/
克鲁斯卡尔算法

#include<bits/stdc++.h>
using namespace std;
const int N=110;
int a[N][N],n,p[N];
struct node{int a,b,c;};
bool cmp(node a,node b) {return a.c<b.c;}
vector<node>ve;
int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
int f()
{
    int res=0;
    for(int i=1;i<=n;i++) p[i]=i;
    sort(ve.begin(),ve.end(),cmp);
    for(int i=0;i<ve.size();i++)
    {
        int a=ve[i].a,b=ve[i].b,c=ve[i].c;
        if(find(a)!=find(b))
        {
            res+=c;
            p[find(a)]=find(b);
        }
    }
    return res;
}
int main(void)
{
    cin>>n;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++) cin>>a[i][j];
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(i==j) continue;
            ve.push_back({i,j,a[i][j]});
            ve.push_back({j,i,a[i][j]});
        }
    }
    cout<<f();
    return 0;
}

普利姆算法

#include<bits/stdc++.h>
using namespace std;
const int N=110;
int a[N][N],st[N],dist[N],n;
int f()
{
    int res=0;
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++) if(!st[j]&&(t==-1 || dist[j]<dist[t])) t=j;
        st[t]=1;
        res+=dist[t];
        for(int j=1;j<=n;j++) dist[j]=min(dist[j],a[t][j]);
    }
    return res;
}
int main(void)
{
    cin>>n;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++) cin>>a[i][j];
    cout<<f();
    return 0;
}