LeetCode——5733. 最近的房间(Closest Room)[困难]——分析及代码（Java）

一、题目

一个酒店里有 n 个房间，这些房间用二维整数数组 rooms 表示，其中 rooms[i] = [roomIdi, sizei] 表示有一个房间号为 roomIdi 的房间且它的面积为 sizei 。每一个房间号 roomIdi 保证是 独一无二 的。

同时给你 k 个查询，用二维数组 queries 表示，其中 queries[j] = [preferredj, minSizej] 。第 j 个查询的答案是满足如下条件的房间 id ：

• 房间的面积 至少 为 minSizej ，且
• abs(id - preferredj) 的值 最小 ，其中 abs(x) 是 x 的绝对值。
• 如果差的绝对值有 相等 的，选择 最小 的 id 。如果 没有满足条件的房间 ，答案为 -1 。

请你返回长度为 k 的数组 answer ，其中 answer[j] 为第 j 个查询的结果。

示例 1：

输入：rooms = [[2,2],[1,2],[3,2]], queries = [[3,1],[3,3],[5,2]]
输出：[3,-1,3]
解释：查询的答案如下：
查询 [3,1] ：房间 3 的面积为 2 ，大于等于 1 ，且号码是最接近 3 的，为 abs(3 - 3) = 0 ，所以答案为 3 。
查询 [3,3] ：没有房间的面积至少为 3 ，所以答案为 -1 。
查询 [5,2] ：房间 3 的面积为 2 ，大于等于 2 ，且号码是最接近 5 的，为 abs(3 - 5) = 2 ，所以答案为 3 。

示例 2：

输入：rooms = [[1,4],[2,3],[3,5],[4,1],[5,2]], queries = [[2,3],[2,4],[2,5]]
输出：[2,1,3]
解释：查询的答案如下：
查询 [2,3] ：房间 2 的面积为 3 ，大于等于 3 ，且号码是最接近的，为 abs(2 - 2) = 0 ，所以答案为 2 。
查询 [2,4] ：房间 1 和 3 的面积都至少为 4 ，答案为 1 因为它房间编号更小。
查询 [2,5] ：房间 3 是唯一面积大于等于 5 的，所以答案为 3 。

提示：

• n == rooms.length
• 1 <= n <= 10^5
• k == queries.length
• 1 <= k <= 10^4
• 1 <= roomIdi, preferredj <= 10^7
• 1 <= sizei, minSizej <= 10^7

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/closest-room
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、分析及代码

1. 离线算法

（1）思路

将房间和查询按面积及面积需求大小重新排序，结合有序集合 TreeSet，从大到小依次添加符合当前要求的房间，并查询目标房间编号。

（2）代码

class Solution {
    public int[] closestRoom(int[][] rooms, int[][] queries) {
        //将房间按面积排序
        Arrays.sort(rooms, new Comparator<int[]>() {
            public int compare(int[] ra, int[] rb) {
                return ra[1] - rb[1];
            }
        });
        int n = queries.length, m = rooms.length;
        //将查询按最小面积需求从大到小排序
        int[][] q = new int[n][3];
        for (int i = 0; i < n; i++) {//扩展queries，添加序号后进行排序
            q[i][0] = i;
            q[i][1] = queries[i][0];
            q[i][2] = queries[i][1];
        }
        Arrays.sort(q, new Comparator<int[]>() {
            public int compare(int[] qa, int[] qb) {
                return qb[2] - qa[2];
            }
        });
        
        TreeSet<Integer> nums = new TreeSet<>();//有序存放当前符合面积要求的房间编号
        int roomAdd = m - 1;//记录当前未加入TreeSet的房间数
        int[] ans = new int[n];//答案数组
        for (int i = 0; i < n; i++) {
            while (roomAdd >= 0 && rooms[roomAdd][1] >= q[i][2])//将符合当前最小面积要求的房间编号加入TreeSet
                nums.add(rooms[roomAdd--][0]);
            int queryId = q[i][0], targetId = q[i][1];
            if (nums.isEmpty())
                ans[queryId] = -1;
            else {
                Integer fl = nums.floor(targetId), ce = nums.ceiling(targetId);//不超过、不小于目标ID的房间编号
                if (fl == null)
                    ans[queryId] = ce;
                else if (ce == null)
                    ans[queryId] = fl;
                else if (targetId - fl <= ce - targetId)
                    ans[queryId] = fl;
                else
                    ans[queryId] = ce;
            }
        }
        return ans;
    }
}

（3）结果

执行用时 ：97 ms，在所有 Java 提交中击败了 100.00% 的用户；
内存消耗 ：94.8 MB，在所有 Java 提交中击败了 100.00% 的用户。
（目前提交用户量不足，暂无排名）

三、其他

暂无。