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()