【Lintcode】1897. Meeting Room III

题目地址：

https://www.lintcode.com/problem/meeting-room-iii/description

给定一个区间组成的数组$A$，每个区间代表一个会议的开始和结束时间。一共有$r$个会议室。再给定若干个询问，问$(a,b)$这个会议是否可以加入。如果两个会议一个结束而另一个恰好在这个时刻开始，这样认为是不冲突的。题目保证$A$里的会议是可以合法安排到$r$个会议室里的。

思路是扫描线。开一个数组$M$，对于会议$(a,b)$，我们让$M[a]$加$1$，然后让$M[b]$减$1$。$A$遍历完之后，$M[x]$代表的是$x$这个时刻有多少个会议同时在进行。如果$M[x]<r$，那么说明这个时刻是可以插入新会议的，否则说明不能，我们可以用一个数组$P$来标记每个时刻是否需要新会议室（也就是是否可以插入新会议，如果需要新会议室说明不能插入），如果不需要，则标记为$0$，否则标记为$1$，那么：$$P[x]=\{\begin{array}{l}0,M[x]<r\\ 1,M[x]\ge r\end{array}$$给定新的会议$(a,b)$问题就转化为，问$[a,a+1,...,b-1]$这些位置上的$P$是否都是$0$，也就是问以$P[a-1]$为结尾的前缀和是否等于以$P[b-1]$结尾的前缀和。所以只需要再缓存一下$P$的前缀和就行了。代码如下：

public class Solution {
    /**
     * @param intervals: the intervals
     * @param rooms:     the sum of rooms
     * @param ask:       the ask
     * @return: true or false of each meeting
     */
    public boolean[] meetingRoomIII(int[][] intervals, int rooms, int[][] ask) {
        // Write your code here.
        int max = 0;
        for (int i = 0; i < intervals.length; i++) {
            max = Math.max(max, intervals[i][1]);
        }
        for (int i = 0; i < ask.length; i++) {
            max = Math.max(max, ask[i][1]);
        }
        
        int[] mark = new int[max + 1], need = new int[max + 1];
        for (int i = 0; i < intervals.length; i++) {
            mark[intervals[i][0]]++;
            mark[intervals[i][1]]--;
        }
        
        // count记录每个时刻是否需要额外的会议室
        int count = 0;
        for (int i = 0; i < mark.length; i++) {
            count += mark[i];
            if (count < rooms) {
                need[i] = 0;
            } else {
                need[i] = 1;
            }
        }
        
        // 求一下need数组的前缀和，sum[i]表示以need[i]为结尾的前缀和
        int[] sum = new int[max + 1];
        for (int i = 1; i < sum.length; i++) {
            sum[i] = sum[i - 1] + need[i];
        }
    
        boolean[] res = new boolean[ask.length];
        for (int i = 0; i < res.length; i++) {
            res[i] = sum[ask[i][0] - 1] == sum[ask[i][1] - 1];
        }
        
        return res;
    }
}

时间复杂度$O(l_A+l_q)$，空间复杂度$O(l_A)$。