找出二叉搜索树中第k小的元素(Java)

题目描述

给出一个二叉搜索树，返回这个二叉搜索树中第k小的元素

解决

代码：

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