Prim算法

#include<iostream>
using namespace std;
#define MAX 100
#define INF 0x7fffffff

int graph[MAX][MAX];
int prim(int graph[][MAX], int n)
{
    int distance[MAX];//用来存数组下标为终点的,以选中(加入MST最小生成树的)为起点的距离
    int nearest[MAX];//存以数组下标为终点的对应的起点
    int i, j, sum = 0, min, vnear;
    for (i = 2; i <= n; i++)
    {
        distance[i] = graph[1][i];//V_1到其他点的距离
        nearest[i] = 1;//因为在一开始的时候,默认起点全部为V_1
    }
    nearest[1] = 0;
    for (i = 2; i <= n; i++)//循环n-1次,让所有的点都加入到nearest中
    {
        min = INF;
        for (j = 2; j <= n; j++)//检查每个顶点是否与Y最近
        {
            if (distance[j] < min&&distance[j] != 0)
            {
                min = distance[j];
                vnear = j;
            }
        }
        cout << 'V' << vnear << "->";//输出最短路径上的所经过的点

        sum += min;
        distance[vnear] = 0;//表面新的点已经加入MST
        for (j = 2; j <= n; j++)
        {
            if (graph[vnear][j] < distance[j])//将已经加入的新点到其他路径全部遍历一遍,如果发现更近的,则更新,大于等于的都不变
            {
                distance[j] = graph[vnear][j];
                nearest[j] = vnear;
            }
        }
    }
    return sum;
}
int main()
{
    int i, j, k, m, n;
    int cost;
    cin >> m >> n;//m=顶点个数,n=边的个数

    //初始化图G
    for (i = 1; i <= m; i++)
    {
        for (j = 1; j <= m; j++)
        {
            graph[i][j] = INF;
        }
    }
    //构建图G
    for (k = 1; k <= n; k++)
    {
        cin >> i >> j >> cost;
        graph[i][j] = cost;
        graph[j][i] = cost;
    }
    //求解最小生成树
    cout << "V1->";
    cost = prim(graph, m);//注意这里传二位参数的方式
    cout << "最小权值得和=" << cost << endl;
    system("pause");
    return 0;
}
### 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、付费专栏及课程。

余额充值