二叉搜索树中第K小的元素(来源力扣)

给定一个二叉搜索树的根节点 root ，和一个整数 k ，请你设计一个算法查找其中第 k 个最小元素（从 1 开始计数）。

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

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

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

方法1：（用哈希表记录子树的结点数）

分析：
如果node.left的节点个数=k-1，则node为第k小的值
如果node.left的节点个数>k-1，则应该在node.left中找
如果node.left的节点个数<k-1，则应该在node.right中找，并且k -= node.left + 1

class Solution {
    Map<TreeNode, Integer> map = new HashMap<>();
    public int kthSmallest(TreeNode root, int k) {
        help(root);
        TreeNode node = root;
        while (node != null) {
            int left = map.getOrDefault(node.left, 0);
            if (left < k - 1) {
                node = node.right;
                k -= left + 1;
            } else if (left == k - 1) {
                break;
            } else {
                node = node.left;
            }
        }
        return node.val;
    }
    private int help(TreeNode root) {
        if(root == null) return 0;
        map.put(root, help(root.left) + help(root.right) + 1);
        return map.get(root);
    }
}

此方法搜索的时间复杂度为 O(H)，其中 H 是树的高度；当树是平衡树时，时间复杂度取得最小值 O(logN)；当树是线性树时，时间复杂度取得最大值 O(N)。

方法2：（平衡二叉树）