Leetcode 1054. 距离相等的条形码

题目

在一个仓库里,有一排条形码,其中第 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;
    }
};
上一篇:1054 The Dominant Color (20 分)


下一篇:1054:三角形判断--信息学一本通(c++)