二叉搜索树第k小元素

二叉搜索树第k小元素


二叉搜索树的中序遍历是一个递增的序列（左根右）

1. 中序遍历完整二叉搜索树，每次将遍历到的节点存入数组中去，找到第k - 1个节点返回即可
   • k是从下标1开始的，存到结果数组res中，但是下标是从0开始的.
   • 这个样例res = [1,2,3,4,5,6]
   • 时间复杂度O(n) n 表示有多少个节点
2. 中序遍历二叉搜索树k次，返回第K次结果;
   时间复杂度：O(k)

var kthSmallest = function(root, k) {
    // 中序遍历 根左右
    const res = [];
    dfs(root); // res = [1, 2, 3, 4, 5]
    return res[k - 1];
    function dfs(root) {
        if (root == null) return;
        dfs(root.left);
        res.push(root.val);
        dfs(root.right);
    }
};

第二中做法

var kthSmallest = function(root, k) {
    var res = null; // 全局变量
    dfs(root);
    return res;
    function dfs(root) {
        if (root === null) return false;
        if (dfs(root.left)) return true;
        if (--k === 0) {
            res = root.val;
            return true;
        }
        return dfs(root.right);
    }
};