Prim 算法

本文介绍了一个使用Prim算法寻找图的最小生成树的C++实现。通过输入顶点数、边数及其权值,程序能够输出最小生成树的各条边及相应的权重。适用于学习图论算法和数据结构的学生及开发者。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

按照课件的思路整合的代码,哈哈

#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;

#define MaxValue 32700
const int NumEdges = 50;         //边条数
const int NumVertices = 10;    //顶点个数
typedef char VertexData;        //顶点数据类型
typedef int EdgeData;              //边上权值类型
typedef struct {
    //VertexData vexList[NumVertices];   //顶点表
    EdgeData Edge[NumVertices][NumVertices];
    //邻接矩阵, 可视为边之间的关系
    int n, e;    //图中当前的顶点个数与边数
} MTGraph;

typedef struct {
    int head,tail;
    int cost;
} MST;

MST T[100];

void Prim ( MTGraph G, MST T[] ,int u ) {
    double  lowcost[G.n];
    int nearvex[G.n];
    for ( int i = 0; i < G.n; i++ ) {
        lowcost[i] = G.Edge[u][i];   //Vu到各点代价
        nearvex[i] = u;                 //及最短带权路径
    }
    nearvex[u] = -1;        //加到生成树顶点集合
    int k = 0;            //存放最小生成树结点的变量
    for ( int i = 0; i < G.n; i++ ) {
        if ( i != u ) {     //循环n-1次, 加入n-1条边
            EdgeData min = MaxValue;
            int v = 0;
            for ( int j = 0; j < G.n; j++ ) {
                if ( nearvex[j] != -1 &&     // =-1不参选
                        lowcost[j] < min ) {
                    v = j;
                    min = lowcost[j];
                }
                //求生成树外顶点到生成树内顶点具有最
                //小权值的边, v是当前具最小权值的边
            }
            if ( v ) {     //v=0表示再也找不到要求顶点
                T[k].tail = nearvex[v]; //选边加入生成树
                T[k].head = v;
                T[k++].cost = lowcost[v];
                nearvex[v] = -1;  //该边加入生成树标记
                for ( int j = 0; j < G.n; j++ )
                    if ( nearvex[j] != -1 &&
                            G.Edge[v][j] < lowcost[j] ) {
                        lowcost[j] = G.Edge[v][j];     //修改
                        nearvex[j] = v;
                    }
            }

        }     //循环n-1次, 加入n-1条边
    }
}

int main() {
    MTGraph G;
    int s,t,path;
    cout<<"请输入顶点个数"<<endl;
    cin>>G.n;
    cout<<"请输入边数"<<endl;
    cin>>G.e;

    for(int i=0; i<G.n; i++) {
        for(int j=0; j<G.n; j++) {
            if(i==j) G.Edge[i][j]=0;
            else G.Edge[i][j]=MaxValue;
        }
    }
    cout<<"请输入各边权值"<<endl;
    for(int i=1; i<=G.e; i++) {
        cin>>s>>t>>path;
        G.Edge[s][t]=G.Edge[t][s]=path;
    }


    Prim(G,T,0);
    cout<<endl;
    for(int i=0;i<G.n-1;i++){
        cout<<T[i].head<<" "<<T[i].tail<<" "<<T[i].cost<<endl;
    } 
    return 0;
}
/*
0 1 28
1 2 16
2 3 12
3 4 22
4 5 25
5 0 10
4 6 24
3 6 18
1 6 14
*/

### Prim算法与Kruskal算法的比较 Prim算法和Kruskal算法都是用于求解加权连通图中最小生成树的经典算法。尽管它们的目标相同,但在实现方式、时间复杂度、空间复杂度以及适用场景等方面存在显著差异。 #### 时间复杂度 - **Prim算法**:在最基础的形式下,Prim算法的时间复杂度为 $O(n^2)$,其中 $n$ 表示顶点的数量。通过使用更高效的数据结构如二叉堆优化后,时间复杂度可以降低至 $O(E\log V)$,这里 $E$ 是边的数量,$V$ 是顶点的数量[^1]。 - **Kruskal算法**:Kruskal算法的时间复杂度主要受到排序所有边的影响,通常为 $O(E\log E)$ 或者等价于 $O(E\log V)$,因为边的数量最多可达 $V(V-1)/2$(对于完全图)[^1]。 #### 空间复杂度 - **Prim算法**:其空间复杂度主要取决于顶点数量,大约为 $O(V)$,因为它需要维护一个包含所有顶点的信息的数据结构来跟踪哪些顶点已经被加入到最小生成树中。 - **Kruskal算法**:其空间复杂度则与边数相关,约为 $O(E)$,主要用于存储所有的边及其权重信息。 #### 实现难度 - **Prim算法**:通常认为Prim算法比Kruskal算法更容易实现,尤其是当使用邻接矩阵作为数据结构时。它依赖于优先队列来选择下一个最近的顶点加入到已有的树中。 - **Kruskal算法**:相比之下,Kruskal算法的实现稍微复杂一些,因为它不仅需要对所有边按权重进行排序,还需要一种机制(如并查集)来检测和避免形成环路。 #### 适用场景 - **Prim算法**:更适合处理边稠密的图,即边的数量接近于顶点数量平方的情况。在这种情况下,Prim算法的性能优势更为明显[^3]。 - **Kruskal算法**:对于稀疏图而言,即边的数量远小于顶点数量平方的情况下,Kruskal算法的表现更加出色。由于只需要对所有边进行一次排序,因此在处理大规模稀疏图时效率更高。 综上所述,虽然两种算法都能有效地找到最小生成树,但根据具体的应用场景选择合适的算法是非常重要的。如果图是稠密的,那么Prim算法可能是更好的选择;而对于稀疏图,则推荐使用Kruskal算法。 ```python # 示例代码 - Kruskal算法的基本框架 class UnionFind: def __init__(self, size): self.parent = list(range(size)) def find(self, x): if self.parent[x] != x: self.parent[x] = self.find(self.parent[x]) return self.parent[x] def union(self, x, y): rootX = self.find(x) rootY = self.find(y) if rootX == rootY: return False self.parent[rootY] = rootX return True def kruskal(n, edges): uf = UnionFind(n) res = [] for u, v, weight in sorted(edges, key=lambda x: x[2]): if uf.union(u, v): res.append((u, v, weight)) if len(res) == n - 1: break return res ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值