题目
给定一个二叉搜索树的根节点 root
,和一个整数 k
,请你设计一个算法查找其中第 k
****个最小元素(从 1 开始计数)。
示例 1:
输入:root = [3,1,4,null,2], k = 1
输出:1
思路
- 首先了解什么是二叉搜索树,所谓二叉搜索树是指,左节点小于根节点,右节点大于根节点
- 二叉树的中序遍历:左→根→右。中序遍历时将节点放进list中,最终会得到一个有序的list
- 根据 index = k-1 取出list中的元素即为目标元素。
实现代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public int kthSmallest(TreeNode root, int k) {
ArrayList<Integer> list = new ArrayList<Integer>();
// 中序遍历
inorder(root,list);
return list.get(k-1);
}
public void inorder(TreeNode root,ArrayList<Integer> list) {
if(root == null) return;
inorder(root.left,list);
list.add(root.val);
inorder(root.right,list);
}
}