LeetCode——230. 二叉搜索树中第K小的元素（Java）

Posted 2021-10-17 21丶01丶15

题目描述

题干：
给定一个二叉搜索树的根节点 root ，和一个整数 k 
请你设计一个算法查找其中第 k 个最小元素（从 1 开始计数）。

示例 1：
输入：root = [3,1,4,null,2], k = 1
输出：1

示例 2：
输入：root = [5,3,6,2,4,null,null,1], k = 3
输出：3

---

题目描述

返回树中第k个元素，根据二叉排序树的性质，中序遍历就是按照大小排序后的结果

这里我们需要第k个元素，就增加一个标志位来终止遍历即可

这里没有采用递归的方法，而是迭代方便终止循环找到第k个元素

---

正确代码

public int kthSmallest(TreeNode root, int k) {
        Deque<TreeNode> stack = new ArrayDeque<>();
        while (root != null || !stack.isEmpty()) {
            while (root != null) {
                stack.push(root);
                root = root.left;
            }
            root = stack.pop();
            --k;
            if (k == 0) {
                break;
            }
            root = root.right;
        }
        return root.val;
    }

---

总结

其实根据二叉排序树的特性来解决是最正规的做法，但是相比一些其他另辟蹊径的方法

还是不太行，比如很多人直接记录节点个数或者采用平衡二叉排序树，直接就是神仙打架了

当时简单题做中序遍历的时候，好像还有一种颜色标记法，有兴趣的可以去看一下

如果文章存在问题或者有更好的题解，欢迎在评论区斧正和评论，各自努力，你我最高处见

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 函数？

 

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

 1 # Definition for a binary tree node.
 2 # class TreeNode(object):
 3 #     def __init__(self, x):
 4 #         self.val = x
 5 #         self.left = None
 6 #         self.right = None
 7 
 8 class Solution(object):
 9     def kthSmallest(self, root, k):
10         """
11         :type root: TreeNode
12         :type k: int
13         :rtype: int
14         """
15         # 先广度优先遍历
16         width = [root]
17         val = [root.val]
18         i = 0
19         while i < len(width):
20             cur = width[i]
21             if cur.left is not None:
22                 width.append(cur.left)
23                 val.append(cur.left.val)
24             if cur.right is not None:
25                 width.append(cur.right)
26                 val.append(cur.right.val)
27             i += 1
28         val.sort()
29         
30         return val[k-1]
31         
32         
33         

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