最小生成树板子

kruskal算法:

#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
struct Node{
    int a,b,val;
    friend bool operator < (const Node& x,const Node& y){
        return x.val< y.val;  //对于sort重载的话,从小到大用小于号
    }
}load[1000];
int sum=0;
int n,m;
int fa[1000];
int cnt=0;
int Find(int a){
    return fa[a]==a ? a : fa[a]=Find(fa[a]);
}
void init(int a,int b){
    fa[Find(b)]=Find(a);
}
void kruskal(){
    for(int i=0;i<m;i++){
        int x,y,z;
        x=load[i].a;
        y=load[i].b;
        z=load[i].val;
        if(Find(x)!=Find(y)){
            init(x,y);
            cnt++;
            sum+=z;
        }
        if(cnt== n-1 ){
            break;
        }
    }
}
int main(){
    cin>>n>>m;
    for(int i=0;i<m;i++){
        cin>>load[i].a>>load[i].b>>load[i].val;
    }
    for(int i=1;i<=n;i++){
        fa[i]=i;
    }
    sort(load,load+m);
    kruskal();
    cout<<sum<<endl;
    return 0;
}

 

prim算法

 

#include<iostream>
#include<stdio.h>
#include<queue>
#include<vector>
#include<stack>
#include<string>
#include<string.h>
#include<algorithm>
using namespace std;
struct Node{
    int d,len;
    friend bool operator < (const Node& a,const Node& b){
        return a.len > b.len; //对于优先队列,从小到大排序用大于号
    }
};
int n,m;
int cnt,sum;
int vis[1000];
vector < Node > v[1000];
priority_queue< Node > q;
void prim(){
    vis[1]=1;
    for(int i=0;i<v[1].size();i++){
        q.push(v[1][i]);
    }
    while(!q.empty()){
        Node now = q.top();
        q.pop();
        if(vis[now.d]){
            continue;
        }
        vis[now.d]=1;
        cnt++;
        sum+=now.len;
        for(int i=0;i<v[now.d].size();i++){
            if(vis[v[now.d][i].d]) continue;
            q.push(v[now.d][i]);
        }
        if(cnt==n-1) break;
    }
}
int main(){
    while(cin>>m>>n){
        if(m==0) break;
        cnt=sum=0;
        while(!q.empty()){
            q.pop();
        }
        for(int i=0;i<=n;i++){
            v[i].clear();
        }
        memset(vis,0,sizeof(vis));
        for(int i=0;i<m;i++){
            int x,y,z;
            cin>>x>>y>>z;
            Node tmp;
            tmp.d=y;
            tmp.len=z;
            v[x].push_back(tmp);
            tmp.d=x;
            v[y].push_back(tmp);
        }
        prim();
        if(cnt==n-1){
            cout<<sum<<endl;
        }else{
            cout<<'?'<<endl;
        }
    }
    return 0;
}

 

posted @ 2019-03-21 16:11  ISGuXing  阅读(274)  评论(0编辑  收藏  举报