给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。 说明: 你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。 示例 1: 输入: root = [3,1,4,null,2], k = 1 3 / \ 1 4 \ 2 输出: 1 示例 2: 输入: root = [5,3,6,2,4,null,null,1], k = 3 5 / \ 3 6 / \ 2 4 / 1 输出: 3 进阶: 如果二叉搜索树经常被修改(插入/删除操作)并且你需要频繁地查找第 k 小的值,你将如何优化 kthSmallest 函数?
题解:
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public int kthSmallest(TreeNode root, int k) { List<Integer> list = new ArrayList<>(); dfs(root, list); return list.get(k-1); } private void dfs(TreeNode node, List<Integer> list) { if (node == null) { return; } dfs(node.left, list); list.add(node.val); dfs(node.right, list); } }
分析: 一开始我的想法是用快慢指针来做的,但是快慢指针,递归起来是有点麻烦,后面看了题解,原来可以用一个数组先存起来,真是秒啊。