二叉搜索树最近节点查询

/*
 * 题目思路：
 * 使用Java Stream对二叉搜索树进行中序遍历，以获得有序的节点列表。
 * 然后对于每个查询，使用二分查找分别找到小于等于和大于等于查询值的最大和最小节点。
 */

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

public class Solution {
    public List<int[]> closestNodes(TreeNode root, int[] queries) {
        List<Integer> nodes = new ArrayList<>();
        inOrderTraversal(root, nodes);
        List<int[]> answer = new ArrayList<>();
        for (int query : queries) {
            int maxVal = nodes.stream().filter(v -> v <= query).reduce((a, b) -> b).orElse(-1);
            int minVal = nodes.stream().filter(v -> v >= query).findFirst().orElse(-1);
            answer.add(new int[] { maxVal, minVal });
        }
        return answer;
    }

    // 中序遍历获得有序节点列表
    private void inOrderTraversal(TreeNode root, List<Integer> nodes) {
        if (root != null) {
            inOrderTraversal(root.left, nodes);
            nodes.add(root.val);
            inOrderTraversal(root.right, nodes);
        }
    }
}