leetcode 二叉搜索树中第K小的元素 python

 

 

 

二叉搜索树中第K小的元素

 

 

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 1

示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 3

进阶：
如果二叉搜索树经常被修改（插入/删除操作）并且你需要频繁地查找第 k 小的值，你将如何优化 kthSmallest 函数？

我的想法是广度优先遍历先构建一个列表，然后排个序

 # Definition for a binary tree node.
 # class TreeNode(object):
 #     def __init__(self, x):
 #         self.val = x
 #         self.left = None
 #         self.right = None
 
 class Solution(object):
     def kthSmallest(self, root, k):
         """
         :type root: TreeNode
         :type k: int
         :rtype: int
         """
         # 先广度优先遍历
         width = [root]
         val = [root.val]
         i = 0
         while i < len(width):
             cur = width[i]
             if cur.left is not None:
                 width.append(cur.left)
                 val.append(cur.left.val)
             if cur.right is not None:
                 width.append(cur.right)
                 val.append(cur.right.val)
             i += 1
         val.sort()
 
         return val[k-1]

如果要频繁插入的时候，就每次插入时候用二分查找 维护这个树内元素的列表