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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:利用中序遍历,二叉搜索树的中序遍历序列是有序的
1.递归:

/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int count = 0;
int res;
public int kthSmallest(TreeNode root, int k) {
inOrder(root,k);
return res;
}

public void inOrder(TreeNode root, int k){
if(root==null||count>k){
return;
}
inOrder(root.left,k);
count++;
if(k==count){
res = root.val;
}
inOrder(root.right,k);
}
}

2.迭代

/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {

public int kthSmallest(TreeNode root, int k) {

Stack<TreeNode> stack = new Stack<>();
int count = 0;
while(!stack.isEmpty()||root!=null){
while(root!=null){
stack.push(root);
root = root.left;
}
root = stack.pop();
count++;
if(count==k){
return root.val;
}
root = root.right;

}
return -1;
}


}