从头做leetcode之leetcode 75 颜色分类

给定一个包含红色、白色和蓝色，一共 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--;
                }
            }
        }
        
    }
};

通过时间：