最小生成树的两种模板

最小生成树的两种模板

Prime算法和Kruskal算法
Prime算法是针对稠密图的，就是边很多，点很少，时间复杂度为o(n²)，就是枚举点，找通路

//从1开始的点
int A[25][25];
int low[25];
int vis[25];
void prim(int n){
    int ans=0;
    int minn,k;
    memset(vis,0,sizeof(vis));
    vis[1]=1;
    for(int i=1; i<=n; i++) low[i]=A[1][i];
    for(int i=1; i<n; i++){
        minn=0x3f3f3f3;
        k=-1;
        for(int j=1; j<=n; j++){
            if(!vis[j] && minn>low[j]){
                minn=low[j];
                k=j;
            }
        }
        if(k == -1) return ;//此时没有最小生成树
        ans+=minn;
        vis[k]=1;
        for(int j=1; j<=n; j++){
            if(!vis[j] && A[k][j]<low[j]) low[j]=A[k][j];
        }
    }
    printf("%d\n",ans);
}

Kruskal算法主要是针对稀疏图，时间复杂度为o(eloge)，算法的证明为贪心的策略

#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn=110;
int e,n;
struct edge{
    int u,v,w;
    int num;
}edges[maxn*2];
bool cmp(edge i,edge j){
    if(i.w != j.w) return i.w<j.w;
    else return i.num<j.num;
}
int pre[maxn];
int Find(int x){
    return x==pre[x]?x:pre[x]=Find(pre[x]);
}
void add(int u,int v){
    int fa=Find(u);
    int fb=Find(v);
    if(fa != fb) pre[fa]=fb;
}
int main(){
    int t;scanf("%d",&t);
    while(t--){
        scanf("%d %d",&n,&e);
        for(int i=0; i<e; i++){
            scanf("%d %d %d",&edges[i].u,&edges[i].v,&edges[i].w);
            edges[i].num=i;
        }
        sort(edges,edges+e,cmp);
        for(int i=1; i<=n; i++) pre[i]=i;
        int sum=0;
        for(int i=0; i<e; i++){
            if(Find(edges[i].u) != Find(edges[i].v)){
                add(edges[i].u,edges[i].v);
                sum+=edges[i].w;
            }
        }
        int cnt=0;
        for(int i=1;i<=n;i++){
            if(pre[i]==i)
            	cnt++;
            if(cnt>1)
            	break;
        }
        if(cnt>1) //如果根节点不只一个则证明此组数据不是一棵树
        printf("?\n");
        else printf("%d\n",sum);
    }
}