【LC】2476. 二叉搜索树最近节点查询

题目描述：

给你一个 二叉搜索树 的根节点 root ，和一个由正整数组成、长度为 n 的数组 queries 。

请你找出一个长度为 n 的 二维 答案数组 answer ，其中 answer[i] = [mini, maxi] ：

mini 是树中小于等于 queries[i] 的 最大值 。如果不存在这样的值，则使用 -1 代替。
maxi 是树中大于等于 queries[i] 的 最小值 。如果不存在这样的值，则使用 -1 代替。
返回数组 answer 。

示例 1 ：

输入：root = [6,2,13,1,4,9,15,null,null,null,null,null,null,14], queries = [2,5,16]
输出：[[2,2],[4,6],[15,-1]]
解释：按下面的描述找出并返回查询的答案：
- 树中小于等于 2 的最大值是 2 ，且大于等于 2 的最小值也是 2 。所以第一个查询的答案是 [2,2] 。
- 树中小于等于 5 的最大值是 4 ，且大于等于 5 的最小值是 6 。所以第二个查询的答案是 [4,6] 。
- 树中小于等于 16 的最大值是 15 ，且大于等于 16 的最小值不存在。所以第三个查询的答案是 [15,-1] 。
示例 2 ：

输入：root = [4,null,9], queries = [3]
输出：[[-1,4]]
解释：树中不存在小于等于 3 的最大值，且大于等于 3 的最小值是 4 。所以查询的答案是 [-1,4]。

提示：

树中节点的数目在范围 [2, 105] 内
1 <= Node.val <= 106
n == queries.length
1 <= n <= 105
1 <= queries[i] <= 106

题解：

class Solution {
    public List<List<Integer>> closestNodes(TreeNode root, List<Integer> queries) {
        List<Integer> list = new ArrayList<Integer>();
        preOrderTraverse(root, list);
        int n = list.size();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++){
            arr[i] = list.get(i);
        }
        Arrays.sort(arr);
        int m = queries.size();
        List<List<Integer>> ans = new ArrayList<List<Integer>>();
        int l = 0, r = n - 1;
        for (int i = 0; i < m; i++){
            List<Integer> t = new ArrayList<Integer>();
            if(arr[0] > queries.get(i)){
                t.add(-1);
                t.add(arr[0]);
                ans.add(t);
                continue;
            }
            if (arr[n - 1] < queries.get(i)){
                t.add(arr[n - 1]);
                t.add(-1);
                ans.add(t);
                continue;
            }
            while (l <= r){
                int mid  = (l + r) >> 1;
                if(arr[mid] <= queries.get(i)){
                    l = mid + 1;
                }else {
                    r = mid - 1;
                }
            }
            t.add(arr[l - 1]); 
            l = 0; r = n - 1;
            while(l <= r){
                int mid = (l + r) >> 1;
                if (arr[mid] >= queries.get(i)){
                    r = mid - 1;
                } else{
                    l = mid + 1;
                }
            }
            t.add(arr[r + 1]);
            ans.add(t);
            l = 0; r = n - 1;
        }
        return ans;
     
    }
    private void preOrderTraverse(TreeNode r, List<Integer> list){
            if(r != null){
                list.add(r.val);
                preOrderTraverse(r.left, list);
                preOrderTraverse(r.right, list);
            }
        }
}