描述
给定一个包含红,白,蓝且长度为 n 的数组,将数组元素进行分类使相同颜色的元素相邻,并按照红、白、蓝的顺序进行排序。
我们可以使用整数 0,1 和 2 分别代表红,白,蓝。
不能使用代码库中的排序函数来解决这个问题。
排序需要在原数组中进行。
样例
给你数组 [1, 0, 1, 2], 需要将该数组原地排序为 [0, 1, 1, 2]。
挑战
一个相当直接的解决方案是使用计数排序扫描2遍的算法。
首先,迭代数组计算 0,1,2 出现的次数,然后依次用 0,1,2 出现的次数去覆盖数组。
给定一个包含红,白,蓝且长度为 n 的数组,将数组元素进行分类使相同颜色的元素相邻,并按照红、白、蓝的顺序进行排序。
我们可以使用整数 0,1 和 2 分别代表红,白,蓝。
不能使用代码库中的排序函数来解决这个问题。
排序需要在原数组中进行。
样例
给你数组 [1, 0, 1, 2], 需要将该数组原地排序为 [0, 1, 1, 2]。
挑战
一个相当直接的解决方案是使用计数排序扫描2遍的算法。
首先,迭代数组计算 0,1,2 出现的次数,然后依次用 0,1,2 出现的次数去覆盖数组。
你否能想出一个仅使用常数级额外空间复杂度且只扫描遍历一遍数组的算法?
分析
对于该问题,可以使用hash表,将数组中的数存入hash表中,这里我们使用的是map(自动key排序),将每一个key以此重新覆盖数组,没覆盖以此将其value减一,直到减为0,然后跳下一个key。
程序
class Solution {
public:
/**
* @param nums: A list of integer which is 0, 1 or 2
* @return: nothing
*/
void sortColors(vector<int> &nums) {
// write your code here
// 方法一:直接排序
//sort(nums.begin(), nums.end());
// 方法二:map修炼
map<int, int> result;
for (int i = 0; i < nums.size(); i++)
result[nums[i]]++;
for (auto &w : result)
cout << "key = " << w.first << " value = " << w.second << endl;
int i = 0;
for (auto &w : result){
while (w.second != 0){
nums[i++] = w.first;
w.second--;
}
}
for (int i = 0; i < nums.size(); i++)
cout << nums[i] << " ";
cout << endl;
}
};