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

    示例 1:
    image.png

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

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

    提示:

    树中的节点数为 n 。
    1 <= k <= n <= 104
    0 <= Node.val <= 104

    进阶:如果二叉搜索树经常被修改(插入/删除操作)并且你需要频繁地查找第 k 小的值,你将如何优化算法?


    1. /**
    2. * Definition for a binary tree node.
    3. * public class TreeNode {
    4. * int val;
    5. * TreeNode left;
    6. * TreeNode right;
    7. * TreeNode() {}
    8. * TreeNode(int val) { this.val = val; }
    9. * TreeNode(int val, TreeNode left, TreeNode right) {
    10. * this.val = val;
    11. * this.left = left;
    12. * this.right = right;
    13. * }
    14. * }
    15. */
    16. class Solution {
    17. int k, res;
    18. public int kthSmallest(TreeNode root, int k) {
    19. this.k = k;
    20. dfs(root);
    21. return res;
    22. }
    23. private void dfs(TreeNode root) {
    24. if (root == null) return;
    25. dfs(root.left);
    26. k--;
    27. if (k == 0) {
    28. res = root.val;
    29. return;
    30. }
    31. dfs(root.right);
    32. }
    33. }