Leetcode刷题笔记1 图论part07

卡码网 53 寻宝

prim算法

prim算法核心就是三步,称为prim三部曲

  1. 第一步,选距离生成树最近节点
  2. 第二步,最近节点加入生成树
  3. 第三步,更新非生成树节点到生成树的距离(即更新minDist数组)
python 复制代码
def prim(v, e, edges):
    import sys
    import heapq
    grid = [[10001] * (v + 1) for _ in range(v + 1)]
 
    for edge in edges:
        x, y, k = edge
        grid[x][y] = k
        grid[y][x] = k
    minDist = [10001] * (v + 1)
    isInTree = [False] * (v + 1)
 
    for i in range(1, v):
        cur = -1
        minVal = sys.maxsize
 
        for j in range(1, v + 1):
            if not isInTree[j] and minDist[j] < minVal:
                minVal = minDist[j] 
                cur = j
 
        isInTree[cur] = True
 
        for j in range(1, v + 1):
            if not isInTree[j] and grid[cur][j] < minDist[j]:
                minDist[j] = grid[cur][j]
 
    result = sum(minDist[2:v + 1])
    return result
 
if __name__ == '__main__':
    import sys
    input = sys.stdin.read
    data = input().split()
    v = int(data[0])
    e = int(data[1])
    index = 2
    edges = []
    for i in range(e):
        x = int(data[index])
        y = int(data[index + 1])
        k = int(data[index + 2])
        edges.append((x, y, k))
        index += 3
    ans = prim(v, e, edges)
    print(ans)

kruskal 算法

kruscal的思路:

  • 边的权值排序,因为要优先选最小的边加入到生成树里
  • 遍历排序后的边
    • 如果边首尾的两个节点在同一个集合,说明如果连上这条边图中会出现环
    • 如果边首尾的两个节点不在同一个集合,加入到最小生成树,并把两节点加入同一个集合
python 复制代码
class Edge:
    def __init__(self, l, r, val):
        self.l = l
        self.r = r
        self.val = val
n = 10001
father = list(range(n))
def init():
    global father 
    father = list(range(n))
def find(u):
    if u != father[u]:
        father[u] = find(father[u])
    return father[u]
def join(u, v):
    u = find(u)
    v = find(v)
    if u != v:
        father[v] = u
 
def kruskal(v, edges):
    edges.sort(key = lambda edge: edge.val)
    init()
    result = 0
    for edge in edges:
        x = find(edge.l)
        y = find(edge.r)
        if x != y:
            result += edge.val
            join(x, y)
    return result
 
if __name__ == '__main__':
    import sys
    input = sys.stdin.read
    data = input().split()
    v = int(data[0])
    e = int(data[1])
    index = 2
    edges = []
    for i in range(e):
        x = int(data[index])
        y = int(data[index + 1])
        k = int(data[index + 2])
        edges.append(Edge(x, y, k))
        index += 3
    ans = kruskal(v, edges)
    print(ans)
相关推荐
爱喝矿泉水的猛男2 小时前
非定长滑动窗口(持续更新)
算法·leetcode·职场和发展
_Kayo_2 小时前
VUE2 学习笔记6 vue数据监测原理
vue.js·笔记·学习
YuTaoShao2 小时前
【LeetCode 热题 100】131. 分割回文串——回溯
java·算法·leetcode·深度优先
使二颗心免于哀伤4 小时前
《设计模式之禅》笔记摘录 - 10.装饰模式
笔记·设计模式
大锦终4 小时前
【算法】前缀和经典例题
算法·leetcode
悠哉悠哉愿意5 小时前
【电赛学习笔记】MaxiCAM 项目实践——与单片机的串口通信
笔记·python·单片机·嵌入式硬件·学习·视觉检测
cccc来财5 小时前
Java实现大根堆与小根堆详解
数据结构·算法·leetcode
岩中竹5 小时前
广东省省考备考——常识:科技常识(持续更新)
笔记
Olrookie6 小时前
若依前后端分离版学习笔记(三)——表结构介绍
笔记·后端·mysql
恣艺7 小时前
LeetCode 854:相似度为 K 的字符串
android·算法·leetcode