1054. 距离相等的条形码
在一个仓库里,有一排条形码,其中第 i 个条形码为 barcodes[i]。
请你重新排列这些条形码,使其中任意两个相邻的条形码不能相等。 你可以返回任何满足该要求的答案,此题保证存在答案。
示例 1:
输入:barcodes = [1,1,1,2,2,2]
输出:[2,1,2,1,2,1]
示例 2:
输入:barcodes = [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/problems/distant-barcodes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码实现:
class Solution {
public:
vector<int> rearrangeBarcodes(vector<int> &barcodes)
{
int n = barcodes.size();
unordered_map<int, int> map;
vector<int> ans(n);
priority_queue<pair<int, int>, vector<pair<int, int>>> heap;
for (int x : barcodes)
{
map[x]++;
}
for (auto x : map)
{
int nn = x.second;
while (nn--)
{
heap.push({x.second, x.first});
}
}
for (int i = 0; i < n; i += 2)
{
ans[i] = heap.top().second;
heap.pop();
}
for (int i = 1; i < n; i += 2)
{
ans[i] = heap.top().second;
heap.pop();
}
return ans;
}
};