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

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

描述

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

说明

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

实例

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

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

进阶

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

题解

• 使用中序遍历,检查当前元素是被遍历到的第几个元素

public static int kthSmallest(TreeNode root, int k) {
        Stack<TreeNode> stack = new Stack<>();
        int counter = 0;

        while (root != null || !stack.isEmpty()){
            while (root != null){
                stack.push(root);
                root = root.left;
            }

            if (!stack.isEmpty()){
                root = stack.pop();
                counter++;
                if (counter == k)
                    return root.val;
                root = root.right;
            }
        }
        return 0;
    }