2022-01-21 每日打卡：难题精刷

2022-01-21 每日打卡：难题精刷

写在前面

“这些事儿在熟练之后，也许就像喝口水一样平淡，但却能给初学者带来巨大的快乐，我一直觉得，能否始终保持如初学者般的热情、专注，决定了在做某件事时能走多远，能做多好。” 该系列文章由python编写，所刷题目共三个来源：之前没做出来的 ；Leetcode中等，困难难度题目； 周赛题目；某个专题的经典题目，所有代码已AC。每日1-3道，随缘剖析，希望风雨无阻，作为勉励自己坚持刷题的记录。

单源最短路

• Dijkstra算法：

import heapq

n, m, s, t = list(map(int, input().split()))
# 注意下标，n个点的下标为 1~n
e = [[]for i in range(n+1)]
dis = [10000000000 for i in range(n+1)]
for i in range(m):
	u, v, w = list(map(int, input().split()))
    # 不要通过循环一遍遍的检查是否该加入队列，会超时
    # 加一遍就是有向边，加两边就是无向边
	e[u].append([v, w])
	e[v].append([u, w])

dis[s] = 0
q = []
# 使用最小堆，默认会按照第一个排序
heapq.heappush(q, [0, s])
while q:
    d, u = heapq.heappop(q)
    # 说明前面已经有更近的u更新过了
    # 且一定加入了堆中
    if d > dis[u]:
        continue
    
    # 更新所有可达路线，而不是选定的路线
    for v, w in e[u]:
        if dis[u]+w < dis[v]:
            dis[v] = dis[u]+w
            # 每次是从「未求出最短路径的点」中 取出 
            # 距离起点！！！ 
            # 最近的点
            heapq.heappush(q, [dis[v], v])


print(dis[t])

最小生成树

• Kruskal:

import heapq


def find(u):
    if parent[u] != u:
        parent[u] = find(parent[u])
    return parent[u]

# 把u并入v的集合


def union(u, v):
    # 这个写法是不可以的，比如(1,5)(1,6)(5,6)
    # 变成(5,5)(6,6)，此时5和6仍不在一个集合
    # parent[u] = find(v)
    # 需要像下面这种，套娃写法！
    # 目的是修改**根**
    parent[find(u)] = find(v)


n, m = list(map(int, input().split()))
# 注意下标，n个点的下标为 1~n
q = []
parent, ans = [_ for _ in range(n+1)], 0
for i in range(m):
    u, v, w = list(map(int, input().split()))
    q.append([w, u, v])

# q.sort(key = lambda x:x[0])
heapq.heapify(q)

# 注意结束条件，n个点的生成树为n-1条边
cnt = 0
while cnt< n-1:
# while q:
    # w,u,v = q.pop(0)
    w, u, v = heapq.heappop(q)
    if find(u) != find(v):
        union(u, v)
        ans += w
        cnt+=1
print(ans)

• Prim：

import heapq

n, m = [ int(a) for a in input().split()]
# 注意下标，n个点的下标为 1~n
e = [[]for i in range(n+1)]
for i in range(m):
	u, v, w = [ int(a) for a in input().split()]
    # 不要通过循环一遍遍的检查是否该加入队列，会超时
    # 加一遍就是有向边，加两边就是无向边
	e[u].append([w, v])
	e[v].append([w, u])

q, tmp = e[1], 1
used, ans = set([1]), 0
heapq.heapify(q)
while len(used) != n:
    w, v = heapq.heappop(q)
    if v not in used:
        used.add(v)
        ans += w
        for i in e[v]:
            heapq.heappush(q, i)
print(ans)