描述
给定一个包含红,白,蓝且长度为 n 的数组,将数组元素进行分类使相同颜色的元素相邻,并按照红、白、蓝的顺序进行排序。
我们可以使用整数 0,1 和 2 分别代表红,白,蓝。
注意事项
不能使用代码库中的排序函数来解决这个问题。
排序需要在原数组中进行。
样例
给你数组 [1, 0, 1, 2], 需要将该数组原地排序为 [0, 1, 1, 2]。
思考
- 各种排序
- 也可以计算各个的数量,再 push_back()
代码
// By Lentitude
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
if (nums.empty()) return;
bool isExist;
for (int i = 0; i != nums.size(); ++i){
isExist = false;
for (int j = 1; j != nums.size() - i; ++j){
if (nums[j - 1] > nums[j] ){
swap(nums[j - 1], nums[j]);
isExist = true;
}
}
if (!isExist){
return;
}
}
}
};