python使用堆/優先隊列

想用python實現一下單源最短路的Dijkstra算法，卻發現不會使用python的優先隊列，於是學習了一下。

學習的博客：https://www.yangyanxing.com/article/priorityqueue_in_python.html

https://blog.csdn.net/liu2012huan/article/details/53264162

 

博主想簡要小結一下，先說明，博主並沒有深刻了解python堆/隊列的實現，沒有認真讀過官方文檔，沒看過源碼什么的。只是博主照着上面兩篇博文照葫蘆畫瓢，給一個最簡單的使用方式而已。

from queue import Queue,PriorityQueue

# 使用heapq實現優先隊列
#定義一個可比較對象
class pqueue:
    def __init__(self,x,y):
        self.x = x
        self.y = y

    def __lt__(self, other):        #定義了<，像C++的重載<運算符
        return self.x<other.x


def main():
    pq = PriorityQueue()
    while 1:
        opt=int(input())
        if opt==0:      #輸入0，結束測試
            break
        if opt==1:      #輸入1，輸入數字對
            x,y=map(int,input().split())
            pq.put(pqueue(x,y))    #插入數據
        if opt==2:      #輸入2，輸出優先隊列隊頭
            tmp=pq.get()
            print(tmp.x,tmp.y)    #提取隊頭


if __name__=="__main__":
    main()

 

然后是用法示例，用python實現Dijkstra算法：

from queue import Queue,PriorityQueue

class pair:
    def __init__(self,x,y):
        self.x = x
        self.y = y

    def __lt__(self, other):     
        return self.y<other.y

class edge:
    def __init__(self,x,y,z):
        self.x = x
        self.y = y
        self.z = z


def Dijkstra(n,s,edges):
    pq = PriorityQueue()
    vis=[0]*(n+1)
    dis=[0x3f3f3f3f]*(n+1)
    dis[s]=0
    pq.put(pair(s,0))
    while not pq.empty():
        now=pq.get()
        if vis[now.x]==1:
            continue
        vis[now.x]=1
        for e in edges[now.x]:
            if dis[now.x]+e.z<dis[e.y]:
                dis[e.y]=dis[now.x]+e.z
                pq.put(pair(e.y,dis[e.y]))
    INF=(1<<31)-1
    for i in range(1,n+1):
        print(dis[i] if dis[i]!=0x3f3f3f3f else INF,end=' ')

def main():
    n,m,s=map(int,input().split())
    edges=[]
    for _ in range(n+1):
        edges.append([])
    for _ in range(m):
        x,y,z=map(int,input().split())
        edges[x].append(edge(x,y,z))
    Dijkstra(n,s,edges)


if __name__=="__main__":
    main()