【模板】最小生成树

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出orz

输入输出格式

输入格式：

第一行包含两个整数N、M，
表示该图共有N个结点和M条无向边。（N<=5000，M<=200000）
接下来M行每行包含三个整数Xi、Yi、Zi，
表示有一条长度为Zi的无向边连接结点Xi、Yi

输出格式：

输出包含一个数，即最小生成树的各边的长度之和；如果该图不连通则输出orz

输入输出样例

输入样例#1：

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出样例#1：

7

代码

Kruskal（基于并查集的算法）

#include<cstdio>
#include<algorithm>
using namespace std;
const int M=200005,N=5005;
int n,m,ans,h[N],fa[N],t,edge;
struct node{
    int u,v,w;
}e[M];
inline int read()
{
    int sum=0; char ch=getchar();
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch<='9'&&ch>='0') sum=sum*10+ch-48,ch=getchar();
    return sum;
} 
bool cmp(node a,node b)
{
    return a.w<b.w;
}
int find(int x)
{
    if(fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}
int main()
{
    int u,v,w;
    n=read(),m=read();
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=m;i++) 
    {
        e[i].u=read(),e[i].v=read(),e[i].w=read();
    }
    sort(e+1,e+m+1,cmp);
    for(int i=1;i<=m;i++)
    {
        int u=e[i].u;
        int v=e[i].v;
        int w=e[i].w;
        if(find(u)==find(v)) continue;
        fa[find(u)]=v;
        edge++;ans+=w;
        if(edge==n-1) 
        {
            printf("%d",ans);
            return 0;
        }
    }
    printf("orz");
    return 0;
}

Prim(基于dj的算法)

#include<cstdio>
using namespace std;
const int MAXN=0x7fffffff;
int d[5005],w[5005][5005],v[5005];
int m,n,x,y,z,min1,t,sum;
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
     {
        scanf("%d%d%d",&x,&y,&z);
        if(w[x][y]==0||w[x][y]>z)
         {
            w[x][y]=z;
            w[y][x]=z;
         }
     }
    for(int i=1;i<=n;i++) d[i]=MAXN;
    d[1]=0;
    for(int i=1;i<=n;i++)
    {
        min1=MAXN;
        for(int j=1;j<=n;j++)
        {
            if(v[j]==0&&d[j]<min1)
            {
                t=j;
                min1=d[j];
             }
         }
        v[t]=1;
        sum=sum+min1;
        for(int j=1;j<=n;j++)
         if(v[j]==0&&w[t][j]!=0&&d[j]>w[t][j])
          d[j]=w[t][j];
     } 
     printf("%d ",sum);
     return 0;
}