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

/**
 * 思路：
 * 利用二叉搜索树的中序遍历的特性，用Java Stream API获取中序遍历的所有元素
 * 然后直接获取第 k 个最小元素。
 */
import java.util.stream.Collectors;
import java.util.List;
 
public class Solution {
    public int kthSmallest(TreeNode root, int k) {
        List<Integer> sortedElements = inorder(root).stream().collect(Collectors.toList());
        return sortedElements.get(k - 1);
    }
    
    private List<Integer> inorder(TreeNode root) {
        if (root == null) return List.of();
        List<Integer> left = inorder(root.left);
        List<Integer> right = inorder(root.right);
        return Stream.concat(Stream.concat(left.stream(), Stream.of(root.val)), right.stream()).collect(Collectors.toList());
    }
}