Python二叉搜索树的第k个结点

本文阅读 1 分钟
首页 Python笔记 正文

中序遍历输出一个序列,然后找到序列中第k个数即可。

  1. '''
  2. 给定一颗二叉搜索树,请找出其中的第k大的结点。例如,
  3. 5
  4. / \
  5. 3 7
  6. /\ /\
  7. 2 4 6 8 中,
  8. 按结点数值大小顺序第三个结点的值为4。
  9. '''
  10. # -*- coding:utf-8 -*-
  11. class TreeNode:
  12. def __init__(self, x):
  13. self.val = x
  14. self.left = None
  15. self.right = None
  16. class Solution:
  17. # 返回对应节点TreeNode
  18. def __init__(self):
  19. self.treeNode = []
  20. def inOrder(self, pRoot):
  21. if len(self.treeNode) < 0:
  22. return None
  23. if pRoot.left:
  24. self.inOrder(pRoot.left)
  25. self.treeNode.append(pRoot)
  26. if pRoot.right:
  27. self.inOrder(pRoot.right)
  28. def KthNode(self, pRoot, k):
  29. if k == 0 or pRoot == None:
  30. return
  31. self.inOrder(pRoot)
  32. if len(self.treeNode) < k:
  33. return None
  34. return self.treeNode[k-1]
  35. def KthNode2(self, pRoot, k):
  36. if k <= 0 or not pRoot:
  37. return None
  38. treeStack, nodesQue = [], []
  39. pNode = pRoot
  40. while pNode or len(treeStack):
  41. while pNode:
  42. treeStack.append(pNode)
  43. pNode = pNode.left
  44. if len(treeStack):
  45. pNode = treeStack.pop()
  46. nodesQue.append(pNode)
  47. pNode = pNode.right
  48. if k > len(nodesQue):
  49. return None
  50. return nodesQue[k-1]
解压密码: detechn或detechn.com

免责声明

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

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

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

Python序列化二叉树
« 上一篇 01-21
Python数据流中的中位数
下一篇 » 01-21

发表评论