给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
两遍扫描
- 用三个变量记录三个颜色的数量,时间复杂度O(2n)
class Solution {
public:
void sortColors(vector<int>& nums) {
int red = 0, white = 0,blue = 0;
for(int i = 0;i < nums.size();i++){
if(nums[i] == 0) red++;
else if(nums[i] == 1) white++;
else if(nums[i] == 2) blue++;
}
int index = 0;
while(red){
nums[index] = 0;
index++;
red--;
}
while(white){
nums[index] = 1;
index++;
white--;
}
while(blue){
nums[index] = 2;
index++;
blue--;
}
}
};
通过时间:
一遍扫描
- 扫描过程中将0放在左边,2放在右边。
class Solution {
public:
void sortColors(vector<int>& nums) {
int left = 0,right = nums.size()-1;
for(int i = 0;i < nums.size();i++){
if(nums[i] == 0){
if(i > left){
swap(nums[i],nums[left]);
left++;
i--;
}
}
if(nums[i] == 2){
if(i < right){
swap(nums[i],nums[right]);
right--;
i--;
}
}
}
}
};
通过时间: