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

题目

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

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

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。**

先是自己写的糟糕的算法：

/**
 * 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 {
    int cnt = 0;
    List<Integer> list = new ArrayList<>();
    public int kthSmallest(TreeNode root, int k) {
        inorder(root);
        return list.get(k-1);
    }
    public void inorder(TreeNode root){
        if (root == null) return;
        inorder(root.left);
        list.add(root.val);
        inorder(root.right);

    }
}

然后是官方的迭代算法，可以在找到之后就不再查找，降低了时间空间复杂度

class Solution {
  public int kthSmallest(TreeNode root, int k) {
    LinkedList<TreeNode> stack = new LinkedList<TreeNode>();

    while (true) {
      while (root != null) {
        stack.add(root);
        root = root.left;
      }
      root = stack.removeLast();  //弹出时是左根右的顺序
      if (--k == 0) return root.val;
      root = root.right;
    }
  }
}