python构建堆

本文阅读 1 分钟
首页 Python笔记 正文
  1. # 构建树实现堆
  2. class BinHeap:
  3. def __init__(self):
  4. self.heapList = [0]
  5. self.currentSize = 0
  6. #插入新结点后必要时交换子节点和父节点的位置保持堆的性质
  7. def percUp(self, i):
  8. while i//2 > 0:
  9. if self.heapList[i] < self.heapList[i//2]:
  10. temp = self.heapList[i//2]
  11. self.heapList[i//2] = self.heapList[i]
  12. self.heapList[i] = temp
  13. i = i//2
  14. # 插入节点
  15. def insert(self, k):
  16. self.heapList.append(k)
  17. self.currentSize += 1
  18. self.percUp(self.currentSize)
  19. # 删除堆顶元素后, 交换堆尾和空堆顶的位置并实现元素的下沉
  20. def percDown(self, i):
  21. while (i*2) <= self.currentSize:
  22. mc = self.minChild(i)
  23. if self.heapList[i] > self.heapList[mc]:
  24. temp = self.heapList[i]
  25. self.heapList[i] = self.heapList[mc]
  26. self.heapList[mc] = temp
  27. i = mc
  28. def minChild(self, i):
  29. if i * 2 + 1 > self.currentSize:
  30. return i * 2
  31. else:
  32. if self.heapList[i*2] < self.heapList[i*2+1]:
  33. return i * 2
  34. else:
  35. return i * 2 + 1
  36. def delMin(self):
  37. retval = self.heapList[1]
  38. self.heapList[1] = self.heapList[self.currentSize]
  39. self.currentSize = self.currentSize - 1
  40. self.heapList.pop()
  41. self.percDown(1)
  42. return retval
  43. def buildHeap(self, alist):
  44. i = len(alist) // 2
  45. self.currentSize = len(alist)
  46. self.heapList = [0] + alist[:]
  47. while (i > 0):
  48. self.percDown(i)
  49. i = i - 1
  50. return self.heapList
  51. H = BinHeap()
  52. print(H.buildHeap([9, 6, 5, 2, 3]))
解压密码: detechn或detechn.com

免责声明

本站所有资源出自互联网收集整理,本站不参与制作,如果侵犯了您的合法权益,请联系本站我们会及时删除。

本站发布资源来源于互联网,可能存在水印或者引流等信息,请用户自行鉴别,做一个有主见和判断力的用户。

本站资源仅供研究、学习交流之用,若使用商业用途,请购买正版授权,否则产生的一切后果将由下载用户自行承担。

python二叉查找树
« 上一篇 01-21
python二分查找
下一篇 » 01-21

发表评论

惪特博客
  • 文章总数:
    18501 篇
  • 评论总数:
    53360 条
  • 标签总数:
    8881 个
  • 总浏览量:
    23385145 次
  • 最后更新:
    4月27日

最多点赞

随便看看

标签TAG