使用代码形式表达prim的伪代码

以下是 Prim 算法的伪代码：

prim(G, s):
    // 初始化
    for each vertex v in G:
        cost[v] = infinity
        parent[v] = null
        visited[v] = false
    cost[s] = 0
    queue.add(s)

    // 执行 Prim 算法
    while queue is not empty:
        u = queue.removeMin()
        visited[u] = true
        for each neighbor v of u:
            if not visited[v] and cost[v] > weight(u, v):
                cost[v] = weight(u, v)
                parent[v] = u
                queue.add(v)

    // 返回最小生成树
    return parent

其中，`G` 是一个无向带权图，`s` 是起始顶点，`cost[v]` 表示从 `s` 到 `v` 的最小距离（初值为无穷大），`parent[v]` 表示 `v` 在最小生成树中的父节点（初值为 `null`），`visited[v]` 表示顶点 `v` 是否已被访问过，`queue` 是一个优先队列，初始时加入起始顶点 `s`。算法的核心是不断从队列中取出距离起始顶点最近的顶点 `u`，然后将 `u` 的邻居顶点 `v` 加入队列，并更新 `cost[v]` 和 `parent[v]`。最终返回的是最小生成树的所有边的起始顶点的数组。