题目描述:
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。

此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

思路1:统计nums数组中0,1,2元素的个数,然后重写nums数组

class Solution {
public:
void sortColors(vector<int>& nums) {
if(nums.empty())
return ;
vector<int> vec(3,0);
//fill(vec.begin(),vec.end(),0);
for(int i=0;i<nums.size();i++){
vec[nums[i]]++;//统计0,1,2的个数
}
int k=0;//用于标记重写后nums的下标
for(int i=0;i<3;i++){
while(vec[i]--){
nums[k++]=i;
}
}
}

};

基本排序算法汇总+LeetCode_75. 颜色分类_i++


思路2:采用常用的排序算法,对nums数组中的元素进行排序

1、冒泡排序

class Solution {
public:
void sortColors(vector<int>& nums) {
if(nums.empty())
return ;
int n=nums.size();
for(int i=n-1;i>0;i--){
bool flag=false;//标记,如果本轮没发生交换,则有序,不需要再排
for(int j=n-1;j>n-1-i;j--){//每次都是从最后往前比较
if(nums[j]<nums[j-1]){
flag=true;
swap(nums[j],nums[j-1]);
}
}
if(flag==false)
return ;
}
}
};

基本排序算法汇总+LeetCode_75. 颜色分类_排序算法_02


未完待续。。。。。。