目录
力扣1054. 距离相等的条形码
解析代码
力扣1054. 距离相等的条形码
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
class Solution {
public:
vector<int> rearrangeBarcodes(vector<int>& barcodes) {
}
};
解析代码
贪心策略:
- 每次处理一批相同的数字,往 n 个空里面摆放。
- 每次摆放的时候,隔一个格子摆放一个数。
- 先处理出现次数最多的那个数,剩下的数可任意。此题保证存在答案->出现次数最多的那个数不会超过(n + 1)/ 2,下一个数想相邻的话只能“填一圈”(不可能)。
class Solution {
public:
vector<int> rearrangeBarcodes(vector<int>& barcodes) {
unordered_map<int, int> hash; // 数和其出现的次数
int mostVal = 0, maxCount = 0;
for(auto& e : barcodes) // 统计每个数出现的频次
{
++hash[e];
if(maxCount < hash[e])
{
maxCount = hash[e];
mostVal = e;
}
}
int n = barcodes.size(), index = 0;
vector<int> ret(n);
for(int i = 0; i < maxCount; ++i) // 先处理出现次数最多的数
{
ret[index] = mostVal;
index += 2;
}
hash.erase(mostVal);
for(auto& [a, b] : hash) // 处理剩下的数
{
for(int i = 0; i < b; ++i)
{
if(index >= n)
index = 1;
ret[index] = a;
index += 2;
}
}
return ret;
}
};