题目描述
给定一个二叉搜索树,编写一个函数 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