最小生成树prim(优先队列优化)算法+Kruskal算法

最小生成树

1.prim算法

算法思想:从任意一点出发,记录点的最小权值,每一次将最小边的结点标记一下,直到所有的点都被加到树里面。优先队列将边按从小到大的顺序排列,队首为最小的边。
板子题:HUD-1863

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <vector>
#include <map>
using namespace std;
const int N=1e5;
int head[N];
int cnt=0;
struct node{
    int to,next,val;
}mp[N];
struct nod{               //优先队列排序;
    int pos,val;
    bool friend operator <(nod a, nod b)
    {
        return a.val>b.val;
    }
};
void add(int x, int y, int val)  //链式向前星建边;
{
    mp[cnt].to=y;
    mp[cnt].val=val;
    mp[cnt].next=head[x];
    head[x]=cnt++;
}
int prime(int m)
{
    bool p[200];
    priority_queue<nod >q;
    memset(p,false,sizeof(p));   //初始化标记数组;
    int i,to,val,ans=0,tot=0;
    q.push(nod{1,0});            //压入起点值;
    while(!q.empty()){
        nod are=q.top();
        q.pop();
        int u=are.pos; 
        if(p[u]) continue;
        p[u]=true;
        ans+=are.val;                //权值加入答案;
        ++tot;                      //记录个数;
        for(i=head[u];~i;i=mp[i].next){
            to=mp[i].to;
            val=mp[i].val;
            q.push(nod{to,val});
        }
    }
    if(tot==m) return ans;
    return 0;
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF&&n){
        int x,y,val;
        cnt=0;
        memset(head,-1,sizeof(head));
        for(int i=0;i<n;++i){
            scanf("%d%d%d",&x,&y,&val);
            add(x,y,val);
            add(y,x,val);
        }
        int ans=prime(m);
        if(ans){
            printf("%d\n",ans);
        }
        else{
            printf("?\n");
        }
    }
    return 0;
}

2.Kruskal算法;

算法思想:
1.先对所有的边排序;
2.依次从小到大将边加入到树中,在维护结点的时候运用并查集判断是否已经在相同集合中,若不再相同结点中则将边加入到树中;

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <vector>
#include <map>
using namespace std;
const int N=1e3;
int pre[N];
struct node{
    int x,y,val;
}mp[N*10];
bool cpu(node a, node b)    //对边排序;
{
    return a.val<b.val;
}
int find(int x)           //并查集查找根节点+路径压缩;
{
    if(pre[x]==x) return pre[x];
    return pre[x]=find(pre[x]);
}
int kruskal(int n, int m)
{
    int tot=0,ans=0,fx,fy,x,y,val;
    for(int i=0;i<n;++i){
        x=mp[i].x;
        y=mp[i].y;
        val=mp[i].val;
        fx=find(x);
        fy=find(y);
        if(tot==m-1) break;
        if(fx!=fy){
            pre[fx]=fy;
            ans+=val;
            ++tot;
        }
    }
    if(tot==m-1 ) return ans;
    return 0;
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF&&n){
        for(int i=0;i<n;++i) {
         scanf("%d%d%d",&mp[i].x,&mp[i].y,&mp[i].val);
        }
        sort(mp,mp+n,cpu);
        for(int i=1;i<=m;++i) pre[i]=i;      //并查集初始化前导节点数组;
        int ans=kruskal(n,m);
        if(ans) printf("%d\n",ans);
        else printf("?\n");
    }
    return 0;
}
  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Prim算法Kruskal算法都是用于求解最小生成树的经典算法Prim算法的基本思想是从一个点开始,每次选择一个与当前生成树距离最近的点加入生成树中,直到所有点都被加入生成树为止。具体实现时,可以使用一个优先队列来维护当前生成树与未加入生成树的点之间的距离,每次选择距离最小的点加入生成树中。 Kruskal算法的基本思想是从边开始,每次选择一条权值最小且不会形成环的边加入生成树中,直到生成树中包含所有点为止。具体实现时,可以使用并查集来判断是否形成环。 下面是Prim算法Kruskal算法的C语言代码实现: Prim算法: ```c #include <stdio.h> #include <stdlib.h> #include <limits.h> #define MAX_VERTICES 1000 int graph[MAX_VERTICES][MAX_VERTICES]; int visited[MAX_VERTICES]; int dist[MAX_VERTICES]; int prim(int n) { int i, j, u, min_dist, min_index, sum = 0; for (i = 0; i < n; i++) { visited[i] = 0; dist[i] = INT_MAX; } dist[0] = 0; for (i = 0; i < n; i++) { min_dist = INT_MAX; for (j = 0; j < n; j++) { if (!visited[j] && dist[j] < min_dist) { min_dist = dist[j]; min_index = j; } } u = min_index; visited[u] = 1; sum += dist[u]; for (j = 0; j < n; j++) { if (!visited[j] && graph[u][j] < dist[j]) { dist[j] = graph[u][j]; } } } return sum; } int main() { int n, m, i, j, u, v, w; scanf("%d%d", &n, &m); for (i = 0; i < n; i++) { for (j = 0; j < n; j++) { graph[i][j] = INT_MAX; } } for (i = 0; i < m; i++) { scanf("%d%d%d", &u, &v, &w); graph[u][v] = graph[v][u] = w; } printf("%d\n", prim(n)); return 0; } ``` Kruskal算法: ```c #include <stdio.h> #include <stdlib.h> #include <limits.h> #define MAX_VERTICES 1000 #define MAX_EDGES 1000000 struct edge { int u, v, w; }; int parent[MAX_VERTICES]; struct edge edges[MAX_EDGES]; int cmp(const void *a, const void *b) { return ((struct edge *)a)->w - ((struct edge *)b)->w; } int find(int x) { if (parent[x] == x) { return x; } return parent[x] = find(parent[x]); } void union_set(int x, int y) { parent[find(x)] = find(y); } int kruskal(int n, int m) { int i, sum = 0; for (i = 0; i < n; i++) { parent[i] = i; } qsort(edges, m, sizeof(struct edge), cmp); for (i = 0; i < m; i++) { if (find(edges[i].u) != find(edges[i].v)) { union_set(edges[i].u, edges[i].v); sum += edges[i].w; } } return sum; } int main() { int n, m, i; scanf("%d%d", &n, &m); for (i = 0; i < m; i++) { scanf("%d%d%d", &edges[i].u, &edges[i].v, &edges[i].w); } printf("%d\n", kruskal(n, m)); return 0; } ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值