使用buildheap时尝试HeapSort

2024-06-01 01:24:52 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在使用buildheap时进行heapsort,但由于某些原因,我的函数无法工作。如果我不把HeapSort函数中的代码放在函数的外部,而不是放在函数的内部,它就可以工作。我不知道如何通过创建HeapSort函数来实现

class BinHeap:
    def __init__(self):
        self.heapList = [0]
        self.currentSize = 0
    def percUp(self,i):
        while i // 2 > 0:
          if self.heapList[i] < self.heapList[i // 2]:
             tmp = self.heapList[i // 2]
             self.heapList[i // 2] = self.heapList[i]
             self.heapList[i] = tmp
          i = i // 2
    def insert(self,k):
      self.heapList.append(k)
      self.currentSize = self.currentSize + 1
      self.percUp(self.currentSize)
    def percDown(self,i):
      while (i * 2) <= self.currentSize:
          mc = self.minChild(i)
          if self.heapList[i] > self.heapList[mc]:
              tmp = self.heapList[i]
              self.heapList[i] = self.heapList[mc]
              self.heapList[mc] = tmp
          i = mc
    def minChild(self,i):
      if i * 2 + 1 > self.currentSize:
          return i * 2
      else:
          if self.heapList[i*2] < self.heapList[i*2+1]:
              return i * 2
          else:
              return i * 2 + 1
    def delMin(self):
      retval = self.heapList[1]
      self.heapList[1] = self.heapList[self.currentSize]
      self.currentSize = self.currentSize - 1
      self.heapList.pop()
      self.percDown(1)
      return retval
    def buildHeap(self,alist):
      i = len(alist) // 2
      self.currentSize = len(alist)
      self.heapList = [0] + alist[:]
      while (i > 0):
          self.percDown(i)
          i = i - 1

    def HeapSort(alist): 
        bh = BinHeap()
        bh.buildHeap(alist)
        while bh.currentSize != 0:
            print(bh.delMin())

te = [9,5,6,2,3]                    
print(HeapSort(te))

Tags: 函数selfreturnifdefmctmpbh
1条回答
网友
1楼 · 发布于 2024-06-01 01:24:52

你有答案,就是不能还?你知道吗

def HeapSort(alist):
    result = []
    bh = BinHeap()
    bh.buildHeap(alist)
    while bh.currentSize != 0:
        result.append(bh.delMin())
    return result

如果您想让它返回一个列表,只需构建一个列表并返回它。你知道吗

相关问题 更多 >