[LeetCode] 1054. Distant Barcodes

In a warehouse, there is a row of barcodes, where the ith barcode is barcodes[i].

Rearrange the barcodes so that no two adjacent barcodes are equal. You may return any answer, and it is guaranteed an answer exists.

Example 1:

Input: barcodes = [1,1,1,2,2,2]
Output: [2,1,2,1,2,1]

Example 2:

Input: barcodes = [1,1,1,1,2,2,3,3]
Output: [1,3,1,3,1,2,1,2] 

Constraints:

• 1 <= barcodes.length <= 10000
• 1 <= barcodes[i] <= 10000

距离相等的条形码。

在一个仓库里，有一排条形码，其中第 i 个条形码为 barcodes[i]。

请你重新排列这些条形码，使其中两个相邻的条形码 不能 相等。 你可以返回任何满足该要求的答案，此题保证存在答案。

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

思路是排序 + 贪心。因为要确保结果数组内不能有相同元素相邻，所以思路一开始一定是优先放出现次数最多的元素，接着是第二多的，第三多的等等。做法是先用 hashmap 统计不同元素分别都出现了多少次，然后创建一个记录元素出现次数的最大堆，把所有元素的 key 和出现次数放入。这样每次优先被弹出的元素一定是出现次数最多的元素。同时我们需要另外一个变量 lastKey 记录上一个被写入结果集的 key 是什么，如果当前出现次数最多的 key 刚刚被写过一次，则从最大堆中再弹出另外一个 key 写入结果集。否则类似如下这个case是过不去的，第一次写入一个 1 之后，1 仍然是出现次数最多的元素。

[1,1,1,1,2,2,3,3]

时间O(nlogn)

空间O(n)

Java实现

class Solution {
    public int[] rearrangeBarcodes(int[] barcodes) {
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int num : barcodes) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }

        // [key, value]
        // maxHeap
        PriorityQueue<int[]> queue = new PriorityQueue<>((a, b) -> b[1] - a[1]);
        for (int key : map.keySet()) {
            queue.offer(new int[] { key, map.get(key) });
        }

        int lastKey = 0;
        List<Integer> res = new ArrayList<>();
        while (!queue.isEmpty()) {
            int[] first = queue.poll();
            int key = first[0];
            int value = first[1];
            if (key != lastKey) {
                res.add(key);
                value--;
                if (value > 0) {
                    queue.offer(new int[] { key, value });
                }
                lastKey = key;
            } else {
                int[] second = queue.poll();
                int secondKey = second[0];
                int secondValue = second[1];
                res.add(secondKey);
                secondValue--;
                if (secondValue > 0) {
                    queue.offer(new int[] { secondKey, secondValue });
                }
                lastKey = secondKey;
                queue.offer(first);
            }
        }
        int[] list = new int[res.size()];
        for (int i = 0; i < list.length; i++) {
            list[i] = res.get(i);
        }
        return list;
    }
}

 

这道题还有一个思路，还是需要用 hashmap 统计每个不同元素的出现次数，但是写入的时候，是先把出现次数最多的元素写入所有偶数的 index，再把次多的元素写入所有奇数的 index。但是这个思路的时间复杂度并没有更好，因为还是需要涉及到出现次数的排序。

LeetCode 题目总结