LeetCode-230-二叉搜索树中第K小的元素

2022-04-17 04:31:06 阅读：182 来源： 互联网

标签：TreeNode 230 中序节点 values 二叉树 root 树中 LeetCode

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

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

示例说明请见LeetCode官网。

来源：力扣（LeetCode）

https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst/

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一：二叉树的中序遍历、

二叉搜索树的特性即中序遍历结果就是节点的顺序排列，所以通过中序遍历的方式来查找第k个最小元素，处理过程如下：

首先判断特殊情况，如果二叉树为空，说明没有第K小的节点，直接返回空；

中序遍历得到二叉搜索树的所有节点，是按顺序排列的；

判断如果k大于二叉树的节点数，说明不存在第K小的节点，直接返回空；

返回第K小的节点。

import com.kaesar.leetcode.TreeNode;

import java.util.ArrayList;
import java.util.List;

public class LeetCode_230 {
    public static int kthSmallest(TreeNode root, int k) {
        // 如果二叉树为空，说明没有第K小的节点，直接返回空
        if (root == null) {
            return -1;
        }
        List<Integer> values = new ArrayList<>();

        // 中序遍历得到二叉搜索树的所有节点，是按顺序的
        inOrder(root, values);

        // 如果k大于二叉树的节点数，说明不存在第K小的节点，直接返回空
        if (k > values.size()) {
            return -1;
        }

        // 返回第K小的节点
        return values.get(k - 1);
    }

    /**
     * 二叉树的中序遍历
     *
     * @param root
     * @param values
     */
    private static void inOrder(TreeNode root, List<Integer> values) {
        if (root.left != null) {
            inOrder(root.left, values);
        }
        values.add(root.val);
        if (root.right != null) {
            inOrder(root.right, values);
        }
    }

    public static void main(String[] args) {
        /**
         * 测试用例：
         *     3
         *   1   4
         *    2
         */
        TreeNode root = new TreeNode(3);
        root.left = new TreeNode(1);
        root.right = new TreeNode(4);
        root.left.right = new TreeNode(2);

        // 期望输出： 1
        System.out.println(kthSmallest(root, 1));
    }
}

　　【每日寄语】我们的希望是今天比昨天好，明天对于我们来说太遥远了，无所谓。人活着一生不能做萤火虫，永远要当一个钻井工人，永远在头上，为了前面的光芒努力，不是看后面怎么样。希望我们的过去再美好也是由于我们过去的努力，今天再努力就是为了明天的辉煌。

标签：TreeNode,230,中序,节点,values,二叉树,root,树中,LeetCode
来源： https://www.cnblogs.com/cgkj/p/16154920.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

LeetCode-230-二叉搜索树中第K小的元素

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