题目
在一个仓库里,有一排条形码,其中第 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
题解
在优先队列中用pair记录数组信息,其中first为某个数字出现的次数,second为对应的数字。用last上次记录在答案中的数字对应的pair,用current记录当前获得的pair,将current对应的数字记录到答案中,若last数字剩余次数大于0则重新保存到优先队列中,然后将current记录保存到last中。
代码
C++
class Solution {
public:
vector<int> rearrangeBarcodes(vector<int>& barcodes) {
int sz = barcodes.size();
map<int, int> mp;
for(int i = 0; i < sz; i++)
{
mp[barcodes[i]]++;
}
priority_queue<pair<int, int>> q;
for(map<int, int>::iterator it = mp.begin(); it != mp.end(); it++)
{
q.push(make_pair(it->second, it->first));
}
vector<int> ans;
pair<int, int> last;
if(q.size() > 0)
{
last = q.top();
q.pop();
ans.push_back(last.second);
last.first--;
}
while(q.size() > 0)
{
pair<int, int> current = q.top();
q.pop();
ans.push_back(current.second);
current.first--;
if(last.first > 0)
{
q.push(last);
}
last = current;
}
return ans;
}
};