刷题日记:二叉搜索树中第K小的元素

题目描述:

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

题解：

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} k
 * @return {number}
 */
var kthSmallest = function(root, k) {
	// 二叉搜索树的中序遍历就是排好序(从小到大排序)的数组
    // let res = null
    // let stack = [];
    // const dfs = function(root){
    //     if(!root) return
    //     dfs(root.left)
    //     stack.push(root)
    //     dfs(root.right)
    // }
    // dfs(root)
    // res = stack[k-1] ? stack[k - 1].val : null
    // return res
    // 优化上面是遍历了全部的节点,找第k小的元素只需要遍历第k个节点即可
    const stack = [];
    while (root != null || stack.length) {
        while (root != null) { // 先将每个节点的左子节点入栈
            stack.push(root);
            root = root.left;
        }
        root = stack.pop(); // 开始遍历 将后入栈的左节点弹出
        k -= 1
        if (k === 0) { // 遍历到第k个 跳出循环
            break;
        }
        root = root.right;
    }
    return root.val;   
};