python中的堆数据结构

Question

我已经在 python 中实现了最小堆，但经过一些测试后，我意识到有时大约在每 10 次插入后，它会将其置于错误的位置（父级大于子级）。 代码：

class minHeap():
    def __init__(self):
        self.heap = [] # main array
        
    def get_parent(self, i):
        return i//2

    def bubble_up(self):
        i = len(self.heap)-1
        p = self.get_parent(i)
        while self.heap[p] > self.heap[i]:
            self.heap[i], self.heap[p] = self.heap[p], self.heap[i] # swapping parent and child
            i = p
            p = self.get_parent(i)

    def insert(self, k):
        self.heap.append(k)
        self.bubble_up()

我最近 2 个小时一直在寻找错误，所以如果有人能帮助我，我将不胜感激。

Answer 1

解决方案

你应该改变这个函数

def get_parent(self, i):
    return i//2

收件人：

def get_parent(self, i):
    return (i-1)//2

---

说明

您正在使用 python 中的列表来初始化 0-indexed。

self.heap = [] # main array

现在，假设您添加了 2 个元素 1 和 5。

self.heap = [1,5]

现在。如果再添加一个元素，即 3。

self.heap = [1,5,3]

现在，3 的索引是 2。它的父级应该是 1(index = 0)。但是 i//2 会给你 5(index = 1) 作为2//2 = 1。

所以，你应该使用(i-1)//2，它会给你0。

(2-1)//2 = 0

---

希望你能理解我的解决方案?。

---