最小生成树模板

Prim

本质是维护一个集合，选择一个点作为初始点A先加入集合，遍历未加入集合的点，将它们到A的距离设为到集合的最短距离，选则其中的最小值，将该点再加入集合。每次有新的点加入集合后，都用该点更新其余未加入集合的点到集合的距离。(与dijkstra不同的是，prim每次选择到集合最近的点，而dijkstra选择到源点最近的点)。

const int INF=1e9;
const int maxn=2e3+5;
int cost[maxn][maxn];
int lowc[maxn],vis[maxn];
ll prim(){
    ll ans=0;
    memset(vis,0,sizeof(vis));
    vis[1]=1;
    for(int i=2;i<=n;i++) lowc[i]=cost[1][i];
    for(int i=2;i<=n;i++){
        int minn=INF,p=-1;
        for(int j=1;j<=n;j++){
            if(!vis[j]&&lowc[j]<minn){
                minn=lowc[j];
                p=j;
            }
        }
        if(minn==INF)return -1;
        ans+=minn;
        vis[p]=1;
        for(int j=1;j<=n;j++){
            if(!vis[j]&&lowc[j]>cost[p][j])
                lowc[j]=cost[p][j];
        }
    }
    return ans;
}

kruskal

建立边结构体，将边按照权值排序，从小到大加入树中，若一条边的两个点已经在同一棵树中(用并查集维护)，则跳过这条边。

struct edge{
    int f,t,w;
}E[maxm];
int fa[maxn];
int find(int x){
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
ll kruskal(){
    ll ans=0;
    sort(E+1,E+1+tot,cmp);
    for(int i=1;i<=n;i++)
        fa[i]=i;
    int cnt=0;
    for(int i=1;i<=tot;i++){
        int u=E[i].f;
        int v=E[i].t;
        u=find(u);v=find(v);//跳到祖先节点
        if(u!=v){
            ans+=E[i].w;
            fa[u]=v;
            cnt++;
        }
        if(cnt==n-1)return ans;
    }
    if(cnt==n-1)return ans;
    else return -1;
}

转载于:https://www.cnblogs.com/ucprer/p/11310123.html