最小生成树模板

最小生成树模板

prim

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e3+20;
const int inf=0x3f3f3f3f;
int G[maxn][maxn];
bool vis[maxn];
int dis[maxn];
int sum,n,m;

void prim(){
    int pos=0;
    sum=0;
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=n;i++) dis[i] = G[1][i];
    vis[1] = true;

    int flag=false,sum=0;
    for(int i=1;i<n;i++){///n个顶点,(n-1)条边
        int maxx = inf;
        for(int j=1;j<=n; j++){
            if( dis[j]<maxx && !vis[j] ){
                maxx = dis[j];
                pos = j;
            }
        }

        if( maxx==inf ){
            flag = true;
            break;
        }

        sum += maxx;
        vis[pos] = true;
        for(int j=1;j<=n;j++){
            if( !vis[j] && dis[j]>G[pos][j] ){
                dis[j] = G[pos][j];
            }
        }
    }
    if( flag==true ) printf("-1\n");
    else printf("%d\n",sum);
}

int main()
{
    while( ~scanf("%d%d",&n,&m) ){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if( i==j ) G[i][j] = 0;
                else G[i][j] = inf;
            }
        }

        while( m-- ){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            if( G[u][v]>w ){
                G[u][v] = w;
                G[v][u] = w;
            }
        }
        prim();
    }
    return 0;
}

 

prim堆优化

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5+10;
const int maxx = 5e3+15;

struct node{
    int to;
    int wi;
    bool operator < (const node &a) const{
        return wi>a.wi;
    }
};

int n,m;
int vis[maxx];
vector<node>e[maxx];
priority_queue<node>q;

ll prim(){
    ll ans=0;
    int cnt=0;
    q.push((node){1,0});
    while( !q.empty() && cnt<=n ){
        node k=q.top();
        q.pop();
        if( vis[k.to] ) continue;
        vis[k.to] = 1;
        ans += k.wi;
        cnt++;
        for(int i=0;i<e[k.to].size();i++){
            if( !vis[e[k.to][i].to] ){
                q.push((node){e[k.to][i].to, e[k.to][i].wi});
            }
        }
    }
    return ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=m; i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        e[u].push_back((node){v,w});
        e[v].push_back((node){u,w});
    }
    printf("%lld\n",prim());
    return 0;
}

 

Kurscal

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5010;
const int maxx = 200100;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;

int fa[maxn];
int dis[maxn][maxn];
int n,m;
struct edge{
    int ui;
    int vi;
    int wi;
}e[maxx];

int find_fa(int u){
    if( fa[u]==u ) return u;
    return fa[u] = find_fa(fa[u]);
}

void Merge(int u, int v){
    int fu = find_fa(u);
    int fv = find_fa(v);

    if( fu!=fv ){
        fa[fv] = fa[fu];
    }
}

void init(){
    for(int i=0;i<=n;i++){
        fa[i] = i;
    }
}

bool cmp(edge a, edge b){
    return a.wi<b.wi;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        e[i].ui = u;
        e[i].vi = v;
        e[i].wi = w;
    }
    sort(e,e+m,cmp);
    init();
    int top=0;
    ll ans=0;
    for(int i=0;i<m;i++){
        int u = e[i].ui;
        int v = e[i].vi;
        int w = e[i].wi;
        if( find_fa(u)!=find_fa(v) ){
            Merge(u,v);
            ans = ans + (ll)w;
            top++;
        }
        if( top==n-1 ) break;
    }

    if( top==n-1 ) printf("%lld\n",ans);
    else printf("orz\n");
    return 0;
}