【C++】数据结构之最小生成树算法

1.公路村村通

现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。
输入格式:

输入数据包括城镇数目正整数N（≤1000）和候选道路数目M（≤3N）；随后的M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见，城镇从1到N编号。
输出格式:

输出村村通需要的最低成本。如果输入数据不足以保证畅通，则输出−1，表示需要建设更多公路。
输入样例:

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例:

12

Prim算法

#include<bits/stdc++.h>
using namespace std;
const int N=1005,INF=0x3f3f3f3f;
int g[N][N];
int dist[3*N];
int n,m;
bool st[N];
int Prim()
{
    int res=0;
    memset(dist,0x3f,sizeof dist);
    for(int i=0;i<n;i++)
    {
        int flag=-1;
        for(int j=1;j<=n;j++)
            if(!st[j]&&(flag==-1||dist[flag]>dist[j]))
                flag=j;
        if(i&&dist[flag]==INF) return INF;
        if(i) res=res+dist[flag];
        st[flag]=true;
        for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[flag][j]);
    }
    return res;
}
int main()
{
    cin>>n>>m;
    memset(g,0x3f,sizeof g);
    while(m--)
    {
        int a,b,c;
        cin>>a>>b>>c;
        g[a][b]=g[b][a]=min(g[a][b],c);
    }
    int t=Prim();
    if(t==0x3f3f3f3f) cout<<-1<<endl;
    else cout<<t<<endl;
    return 0;
}

kruskal算法

#include<bits/stdc++.h>
using namespace std;
const int N=1005;
int p[N];
int n,m;
struct node
{
    int a,b,c;
};
int find(int x)
{
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
bool cmp(node x,node y)
{
    return x.c<y.c;
}
int main()
{
    node d[N];
    int sum=0;
    cin>>n>>m;
    for(int i=1;i<=n;i++) p[i]=i;
    for(int i=0;i<m;i++)
    {
        cin>>d[i].a>>d[i].b>>d[i].c;
    }
    sort(d,d+m,cmp);
    for(int i=0;i<m;i++)
    {
        int a1,a2;
        a1=find(d[i].a);a2=find(d[i].b);
        if(a1!=a2) {p[a1]=a2;sum=sum+d[i].c;}
    }
    int cnt=0;
    for(int i=1;i<=n;i++)
        if(p[i]==i)
            cnt++;
    if(cnt>1) puts("-1");
    else cout<<sum<<endl;
    return 0;
}