使用Python实现优先级队列（堆）--以最大堆为例

思路：
使用list实现

优先级队列的特点就是
1.父节点总是比子节点要大
2.是一种完全二叉树

按照以上特点
1.可以得到父节点的两个子节点的index
2.通过简单的对比大小，交换list中元素的位置，最后得到重新排序的新列表

class Pq(object):
    def __init__(self, arr):
        self.arr = arr#  接受一个列表
        self.index = 0
        self.flag = 1
        while self.flag == 1:
            self.build_q()


    def build_q(self):
        self.flag = 0#  先置0，如果下面经过swap，则在swap函数中又会变为1
        for index in range(len(self.arr)): #  遍历以所有节点为head的三角形最小树
            h = index
            l = index*2 + 1
            r = index*2 + 2#  根据优先级队列的原理，父节点与的两个子节点的index
            if r <= len(self.arr)-1:#  目前节点有两个孩子
                self.shift_up(h, l, r) 
            elif l == len(self.arr)-1:#  目前节点只有左孩子
                if self.arr[h] < self.arr[l]:
                    self.swap(h, l)
            elif l > len(self.arr)-1:#  目前节点已经没有子节点
                continue

    def shift_up(self, head, left, right):#  接受三个元素的列表
        if self.arr[head] < self.arr[left]:
            self.swap(head, left)
        if self.arr[head] < self.arr[right]:
            self.swap(head, right)
    
    def swap(self, index1, index2):#  交换一个列表中，对应下标的两个元素的位置
        self.flag = 1#  交换过就表示形成的树不是最终形态
        temp = self.arr[index1]
        self.arr[index1] = self.arr[index2]
        self.arr[index2] = temp
        
    def incert(self, val):
        self.arr.append(val)
        index = len(self.arr)-1#  最末元素的index
        h_index = 1#  初始化一个h_index，代表插入节点的父节点的index，大于0进入while即可
        while h_index != 0:
            if index % 2 == 0:#  如果，能被2整除，说明是挂在右边，如果不能则是左边
                h_index = (index-1)//2
            else:#  挂在左边
                h_index = index//2
            if self.arr[h_index] < self.arr[index]:
                self.swap(h_index, index)
                index = h_index
    
    def pop(self):
        self.arr[0] = self.arr[-1]
        self.arr.pop()
        self.flag = 1
        while self.flag == 1:
            self.build_q()
        
        
    def show(self):
        print(self.arr)


q = Pq([1,1,2,2,3,3,4])
q.show()
q.incert(5)
q.show()
q.pop()
q.show()


>>>[4, 2, 3, 1, 2, 1, 3]
>>>[5, 4, 3, 2, 2, 1, 3, 1]
>>>[4, 2, 3, 1, 2, 1, 3]

本例程到目前只实现了实现优先级队列的基本功能。