Leetcode 632

給定k個遞增的列表

找到最短的區間包含這k個列表裏每個列表至少一個數

 

 

貪心地去想，顯然這k個數相鄰越近，數字越小。那麼我們首先拿k個列表最小數，看看答案是多少（最大值-最小值就是答案），然後把最小那個數逐次增大，比如這個數是第一個列表第一個數，那麼就替換第一個列表第二個數，然後判答案是否更優，依次比下去。

實現的話可以採用heap維護一個大小爲k的堆來做

    public int[] smallestRange(List<List<Integer>> nums) {
        PriorityQueue<Node> pq = new PriorityQueue<>(new Comparator<Node>() {
            @Override
            public int compare(Node o1, Node o2) {
                return Integer.compare(o1.val, o2.val);
            }
        });
        int len = nums.size();
        int res = Integer.MAX_VALUE;

        //當前隊列裏的最大值
        int max = Integer.MIN_VALUE;
        int minNum = Integer.MIN_VALUE;
        int maxNum = Integer.MIN_VALUE;
        for (int i = 0; i < len; i++) {
            int x = nums.get(i).get(0);
            pq.add(new Node(i, 0, x));
            max = Math.max(max, x);
        }

        while (pq.size() == len) {
            Node mi = pq.poll();
            if (max - mi.val < res) {
                res = max - mi.val;
                minNum = mi.val;
                maxNum = max;
            }
            if (mi.index < nums.get(mi.group).size() - 1) {
                int x = nums.get(mi.group).get(mi.index + 1);
                pq.add(new Node(mi.group, mi.index + 1, x));
                max = Math.max(max, x);
            }
        }
        return new int[]{minNum, maxNum};
    }


    class Node {
        int group;
        int index;
        int val;

        public Node(int group, int index, int val) {
            this.group = group;
            this.index = index;
            this.val = val;
        }
    }