无向图最小生成树

版权声明:去他的版权声明,随便转。互联网本该自由开放 https://blog.csdn.net/ctsas/article/details/54798994

问题

N个点M条边的无向连通图,每条边有一个权值,求该图的最小生成树。

Input

第1行:2个数N,M中间用空格分隔,N为点的数量,M为边的数量。(2 <= N <= 1000, 1 <= M <= 50000)
第2 - M + 1行:每行3个数S E W,分别表示M条边的2个顶点及权值。(1 <= S, E <= N,1 <= W <= 10000)

Output

输出最小生成树的所有边的权值之和。

题解

prim 优先队列+邻接链表

//62 ms 3584 KB

#include<stdio.h>
#include<queue>
#include<vector>
#define INF 0x3f3f3f3f
#define MAX_V 1002
#define MAX_E 50002
using namespace std;
typedef pair<int,int> P;//fiest->cost  second->id
struct edge{int to,cost;};
vector<edge> G[MAX_V];
int V,E;
int used[MAX_V];

int prim(){
    fill(used,used+V+1,false);
    priority_queue<P,vector<P>,greater<P> > que;
    que.push({0,1});
    int res=0;

    while(!que.empty()){
        P p=que.top();que.pop();
        int v=p.first,f=p.second;
        if(used[f]) continue;
        used[f]=true;
        res+=v;

        for(int i=0;i<G[f].size();i++)
            if(!used[G[f][i].to])
                que.push({G[f][i].cost,G[f][i].to});
    }
    return res;
}

int main()
{
    int x,y,v;
    while(~scanf("%d%d",&V,&E)){
        for(int i=0;i<E;i++){
            scanf("%d%d%d",&x,&y,&v);
            G[x].push_back({y,v});
            G[y].push_back({x,v});
        }
        printf("%d\n",prim());
        for(int i=1;i<=V;i++) G[i].clear();
    }
    return 0;
}

Kruskal

//78 ms 2312 KB

#include<stdio.h>
#include<algorithm>
#define INF 0x3f3f3f3f
#define MAX_V 1002
#define MAX_E 50002
using namespace std;
struct edge{int from,to,cost;};
edge es[MAX_E];
int par[MAX_V];
int E,V;

void init(int n){for(int i=0;i<=n;i++) par[i]=i;}
int find(int x){return x==par[x]?x:par[x]=find(par[x]);}
void unite(int x,int y){par[find(y)]=find(x);}
bool same(int x,int y){return find(x)==find(y);}

bool cmp(edge x,edge y){return x.cost<y.cost;};

int kruskal(){
    sort(es,es+E,cmp);
    init(V);
    int res=0;

    for(int i=0;i<E;i++){
        int x=es[i].from,y=es[i].to;
        if(!same(x,y)){
            unite(x,y);
            res+=es[i].cost;
        }
    }
    return res;
}

int main()
{
    while(~scanf("%d%d",&V,&E)){
        for(int i=0;i<E;i++)
            scanf("%d%d%d",&es[i].from,&es[i].to,&es[i].cost);
        printf("%d\n",kruskal());
    }
    return 0;
}
阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页