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

题干

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

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

示例 1:

输入: root = [3,1,4,null,2], k = 1

   3
  / \
 1   4
  \
   2

输出: 1
示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3

       5
      / \
     3   6
    / \
   2   4
  /
 1

输出: 3
进阶：
如果二叉搜索树经常被修改（插入/删除操作）并且你需要频繁地查找第 k 小的值，你将如何优化 kthSmallest 函数？

想法

我们都知道二叉搜索树的排序情况就等于它中序遍历以后的结果，所以写一个中序遍历的递归函数。
再调用一下，取第k-1位置的树就ok

Java 代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public void inorderTraverse(TreeNode root,List list){
        if(root!=null){
            inorderTraverse(root.left,list);//左
            list.add(root.val);//根
             inorderTraverse(root.right,list);//右
            
        }
    }
    public int kthSmallest(TreeNode root, int k) {
        if(root==null){//空判断
            return 0;
        }
        List<Integer> list=new ArrayList<Integer>();
        inorderTraverse(root,list);
        return list.get(k-1);//第k小是k-1位置
        
    }
}

佬码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private int count = 0;
    private int sum = 0;
    public int kthSmallest(TreeNode root, int k) {
        this.count = k;
        f(root);
        return this.sum;
    }
    public void f(TreeNode root)
    {
        if(root == null || count <= 0) return;
        
        f(root.left);

        if(--count == 0)
        {
            this.sum=root.val;
            return;
        }
        f(root.right);
    }
}

总结

树分清成最小情况其实都挺简单的