堆结构常见面试题-最大线段重合问题

1. 题目描述

        给一组线段，判断该组线段重合的最大数量，例如：[1, 2]，[2, 3]不算重合。

2. 核心思路

        首先对线段数组，按线段开始位置从小到大进行排序，然后对排好的数组进行遍历。在遍历过程中，先检查小根堆中是否有小于或等于线段开始位置的数(空的时候直接将末尾压入就行)，若有将其弹出，然后将线段结束位置压入小根堆，并记录每次该流程后小根堆内数的数量。每次循环后记录小根堆内数量中数量最大值就是最大线段重合数。

3. 关键问题

        a. 为什么要对线段数组进行排序?

        答：后面的数大于前面的数确确保了，小根堆中小于前面的一定小于后面的，不会造成个数的少算。

        b. 为什么每次循环后记录小根堆内数量中数量最大值就是最大线段重合数?

        答：小跟堆中留下的数为当前遍历到线段前面所有能够穿过该线段的线段条数，也就是和其重合的条数。

4. 示例代码

private static class Line{
        private int start;
        private int end;

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

        @Override
        public String toString() {
            return "Line{" +
                    "start=" + start +
                    ", end=" + end +
                    '}';
        }
    }
    private static class MyComparator implements Comparator<Line> {

        @Override
        public int compare(Line o1, Line o2) {
            return o1.start - o2.start;
        }
    }
 

//lines是一个二维数组列数为2，零位置为线段起始，1位置为线段末尾
    public static int maxCover(int[][] lines){
        //将线段起始位置包装成一个类，方便排序
        ArrayList<Line> lineArrayList = new ArrayList<>();
        //小根堆
        PriorityQueue<Integer> heap = new PriorityQueue<>();
        int max = 0;

        for(int i = 0; i < lines.length; i++){
            Line line = new Line(lines[i][0], lines[i][1]);
            lineArrayList.add(line);
        }
        //对线段进行排序
        lineArrayList.sort(new MyComparator());

        for(int i = 0; i < lineArrayList.size(); i++){
            //若小根堆内的元素小于线段起始位置则弹出
            if(!heap.isEmpty() && heap.peek() <= lineArrayList.get(i).start){
                heap.poll();
            }
            heap.add(lineArrayList.get(i).end);
            max = heap.size() > max ? heap.size() : max;

        }
        return max;
    }