【Leetcode】759. Employee Free Time

题目地址：

https://leetcode.com/problems/employee-free-time/

给定若干员工的工作时间的区间，每个员工的工作时间区间彼此不相交。要求返回所有员工都不工作的时间区间。长度无穷大的区间不用返回，长度为$0$的区间也不用返回。

法1：扫描线。将每个时间点包装成一个长度为$2$的数组，第$0$位存时间点，第$1$位存”开工“还是”收工“，”开工“记为$0$，”收工“记为$1$（之所以这么规定是因为长度为$0$的区间不能算。这样按照第二关键字排序的时候，开工就会排在前面，避免了将长度为$0$的区间算进去。具体可以看代码）。接着将这些pair排序，时间前者优先；若时间相同则开工优先。接着遍历所有的pair，记录同时在工作的员工数量。如果数量是$0$，说明free time开始，将该区间加入答案并记录free time开始，此时区间左端点就确定了；如果数量不是$0$并且free time是开始状态，说明有人开工了，此时区间右端点就确定了，同时记录free time结束。注意到一开始的排序方式，同一个时间是开工排在前，收工排在后，这样即使那个时刻事实在工作的人数是$0$，由于区间长度是$0$，所以也不会被计入答案，但是在最后最后的时间点，count也是$0$，但这时候是全体员工都收工了，这个区间应该去掉。所以最后的最后，还需要把答案中的最后一个区间删掉。代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution {
    public List<Interval> employeeFreeTime(List<List<Interval>> schedule) {
        List<Interval> res = new ArrayList<>();
        
        List<int[]> list = new ArrayList<>();
        for (List<Interval> intervals : schedule) {
            for (Interval interval : intervals) {
                list.add(new int[]{interval.start, 0});
                list.add(new int[]{interval.end, 1});
            }
        }
        
        list.sort((p1, p2) -> p1[0] != p2[0] ? Integer.compare(p1[0], p2[0]) : Integer.compare(p1[1], p2[1]));
        int count = 0;
        boolean begin = false;
        for (int[] pair : list) {
            if (pair[1] == 0) {
                count++;
            } else {
                count--;
            }
            
            if (count == 0) {
                res.add(new Interval(pair[0], pair[0]));
                begin = true;
            } else if (begin) {
                res.get(res.size() - 1).end = pair[0];
                begin = false;
            }
        }
        
        res.remove(res.size() - 1);
        return res;
    }
}

class Interval {
    public int start;
    public int end;
    
    public Interval() {
    }
    
    public Interval(int start, int end) {
        this.start = start;
        this.end = end;
    }
}

时间复杂度$O(n\log n)$，空间$O(n)$，$n$是区间总数。

法2：TreeMap + 扫描线。将每个时间点对应一个”人数“。如果是开工，就加$1$，否则就减$1$。余下的步骤和上面类似，也是用一个变量记录当前时间开工的人数，逻辑都和上面类似。这里只需注意如果一个时间点对应的人数是$0$，说明同时有相等的人在开工和收工，这个时间点要略过。这个方法用TreeMap的话，可以实现自动排序，并且空间消耗要小很多（虽然空间复杂度是一样的，都是$O(n)$级别）。代码如下：

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

public class Solution {
    public List<Interval> employeeFreeTime(List<List<Interval>> schedule) {
        List<Interval> res = new ArrayList<>();
        
        Map<Integer, Integer> map = new TreeMap<>();
        for (List<Interval> intervals : schedule) {
            for (Interval interval : intervals) {
                map.put(interval.start, map.getOrDefault(interval.start, 0) + 1);
                map.put(interval.end, map.getOrDefault(interval.end, 0) - 1);
            }
        }
    
        int count = 0;
        boolean begin = false;
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (entry.getValue() == 0) {
                continue;
            }
            
            int time = entry.getKey();
            count += entry.getValue();
            if (count == 0) {
                res.add(new Interval(time, time));
                begin = true;
            } else if (begin) {
                res.get(res.size() - 1).end = time;
                begin = false;
            }
        }
        
        res.remove(res.size() - 1);
        return res;
    }
}

时空复杂度一样。