给定一个二叉搜索树的根节点
root
,和一个整数k
,请你设计一个算法查找其中第k
小的元素(从 1 开始计数)。示例 1:
输入:root = [3,1,4,null,2], k = 1 输出:1
示例 2:
输入:root = [5,3,6,2,4,null,null,1], k = 3 输出:3
提示:
- 树中的节点数为
n
。1 <= k <= n <= 104
0 <= Node.val <= 104
javascript/** * 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) { const stack = []; while (root != null || stack.length) { while (root != null) { stack.push(root); root = root.left; } root = stack.pop(); --k; if (k === 0) { break; } root = root.right; } return root.val; };
二叉搜索树的特点:一种特殊的二叉树,其中任何节点的 左子树 上所有节点的值都 小于 该节点的值,而 右子树 上所有节点的值都 大于该节点的值
二叉树的 前/中/后序 遍历,都是相对于 遍历根节点 讲的:
前 序遍历: 根 节点,左子树,右子树
中 序遍历:左子树, 根 节点,右子树
后 序遍历:左子树,右子树, 根节点
因为二叉搜索树和中序遍历的性质,所以二叉搜索树的中序遍历是按照键增加的顺序进行的。于是,我们可以通过中序遍历找到第 k 个最小元素。