爽歪歪666
不以物喜,不以己悲,努力才是永恒的主题。

题目描述

给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8)    中,按结点数值大小顺序第三小结点的值为4。
思路:利用树的中序遍历即可
 1 # -*- coding:utf-8 -*-
 2 # class TreeNode:
 3 #     def __init__(self, x):
 4 #         self.val = x
 5 #         self.left = None
 6 #         self.right = None
 7 class Solution:
 8     # 返回对应节点TreeNode
 9     def KthNode(self, pRoot, k):
10         # write code here
11         # 按照中序遍历
12         result = []
13         if not pRoot:
14             return None
15         result = self.mid(pRoot,result)
16         if k>len(result) or k<=0:
17             return None
18         return result[k-1]
19     def mid(self,pRoot,result):
20         if pRoot:
21             self.mid(pRoot.left,result)
22             result.append(pRoot)
23             self.mid(pRoot.right,result)
24         return result
25         

note:

找节点

result.append(pRoot)

如果只找值,则

result.append(pRoot.val)

 
posted on 2020-06-20 19:01  爽歪歪666  阅读(334)  评论(0编辑  收藏  举报