Leetcode 1054 距离相等的条形码 (C++题解)

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

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

 

示例 1：

输入：[1,1,1,2,2,2]
输出：[2,1,2,1,2,1]
示例 2：

输入：[1,1,1,1,2,2,3,3]
输出：[1,3,1,3,2,1,2,1]
 

提示：

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

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

题解：先使用map记录每个数字出现的次数，根据次数放入优先队列，然后将最多的数字放入数组奇数位，然后将其他数组依次放入奇数位，直到奇数位放满，在将其他数字依次放入偶数位即可。

class Solution {
public:
    vector<int> rearrangeBarcodes(vector<int>& barcodes) {
        int N = barcodes.size();
        map<int, int> m;
        for (auto i : barcodes) ++m[i];
        priority_queue<pair<int, int> > pq;
        for (auto& p : m) pq.push({p.second, p.first});
        vector<int> res(N);
        int i = 0;
        while (!pq.empty()) {
            auto p = pq.top();
            pq.pop();
            for (int j = 0; j < p.first; ++j) {
                res[i] = p.second;
                i = (i + 2 >= N) ? 1 : i + 2; // 隔一位填充一个
            }
        }
        return res;
    }
};