230. 二叉搜索树中第K小的元素 (Java解题 中序遍历)

---

题目描述

难度：中等

给定一个二叉搜索树的根节点 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 <= 10^4
0 <= Node.val <= 10^4

题解分析

已知是二叉搜索树，那么如果我们按照中序遍历的方式遍历二叉树的话，所有元素的访问顺序就是从小到大，我们可以将所有访问到的元素放入 List 中直接返回第k个，也可以在访问的时候进行限制，每访问到一个 k 就减1 ，直到 k 等于0就代表当前这个节点就是第 k 个节点。

解题代码

	/*
        中序遍历
     */
    public int kthSmallest(TreeNode root, int k) {
        Deque<TreeNode> stack = new LinkedList<>();	// 栈
        TreeNode node = root;
        while (true){
            if(node != null){
                stack.offerFirst(node);
                node = node.left;
            } else {
                node = stack.pollFirst();       // 每访问一个元素 k就-1 当k = 0 时，代表访问到 k 了
                if(--k == 0) return node.val;
                node = node.right;
            }
        }
    }

---

总结

本题需要理解二叉搜索树的中序遍历是按照从小到大的顺序来的，这样解题的时候就很容易想到利用中序遍历来写。



岁月悠悠，衰微只及肌肤；热忱抛却，颓废必致灵魂