二叉搜寻树中第 K 小的元素
题目形容:给定一个二叉搜寻树的根节点
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));
}
}
【每日寄语】 咱们的心愿是明天比昨天好,今天对于咱们来说太边远了,无所谓。人活着毕生不能做萤火虫,永远要当一个钻井工人,永远在头上,为了后面的光辉致力,不是看前面怎么样。心愿咱们的过来再美妙也是因为咱们过来的致力,明天再致力就是为了今天的辉煌。