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

题目描述

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

题意理解

1、题意理解：可以简单的理解为，将二叉搜索树排好序，一个排好序的数组，其k是该数组的索引
2、初步想法：用中序遍历把二叉搜索树排好序，再根据k来查找（要怎么找到第k个呢）
从栈中每弹出一个数，变量k减一，直到k为0时，该数即为所求。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public int kthSmallest(TreeNode root, int k) {

        Stack<TreeNode> st = new Stack<>();

        while(root != null){
            st.push(root);
            root = root.left;
        }

        while(k != 0){
            TreeNode n = st.pop();
            k--;
            if(k == 0) return n.val;
            TreeNode right = n.right;
            while(right != null){
                st.push(right);
                right = right.left;
            }
        }
        return -1;
    }
}

参考：https://leetcode.com/problems/kth-smallest-element-in-a-bst/discuss/63660/3-ways-implemented-in-JAVA-(Python)%3A-Binary-Search-in-order-iterative-and-recursive