1054. 距离相等的条形码

转载请声明地址四元君

**1054. 距离相等的条形码 **

题目难度 Medium

在一个仓库里，有一排条形码，其中第 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

思路

一开始直接用回溯法，但是时间超了。

后来发现根本不用这么复杂…直接插空。先计算每个数字有多少个，记录下来，然后开始插空，插两遍。每一遍都是隔一个插一个。

代码

class Solution {
public:
    vector<int> rearrangeBarcodes(vector<int>& barcodes) {
        vector<int> temp(10001,0);
        for(int n:barcodes) temp[n]++;
        vector<int> nums(10001);
        for(int i=0;i<10001;i++) nums[i]=i;
        sort(nums.begin(),nums.end(),[&temp](int a,int b){
            return temp[a]>temp[b];
        });
        
        int step=0;
        int i=0;
        while(step<barcodes.size()){
            if(temp[nums[i]]>0){
                barcodes[step]=nums[i];
                temp[nums[i]]--;
                step+=2;
            }else{
                i++;
            }
        }
        
        step=1;
         while(step<barcodes.size()){
            if(temp[nums[i]]>0){
                barcodes[step]=nums[i];
                temp[nums[i]]--;
                step+=2;
            }else{
                i++;
            }
        }
        return barcodes;
    }
};