关于java:LeetCode230二叉搜索树中第K小的元素

7次阅读

共计 1268 个字符，预计需要花费 4 分钟才能阅读完成。

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

示例阐明请见 LeetCode 官网。

起源：力扣（LeetCode）
链接：https://leetcode-cn.com/probl…
著作权归领扣网络所有。商业转载请分割官网受权，非商业转载请注明出处。

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

二叉搜寻树的个性即中序遍历后果就是节点的顺序排列，所以通过中序遍历的形式来查找第 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));
    }
}