【数组】75. 颜色分类

题目：

 

 

解答：

（1）计数排序，时间复杂度为O(n)，空间复杂度为O(n)。
（2）三个指针的方式，时间复杂度为O(n)，空间复杂度为O(1)。
    用 i 记录0应该放的位置，j 记录2应该放的位置。
    cur从0到j扫描，
    遇到0，放在 i 位置，i 后移
    遇到2，放在 j 位置，j 前移
    遇到1，cur后移
    扫描一遍得到排好序的数组。
    时间O(n)且一次扫描，空间O(1)，满足要求。
    初始值：i = 0， cur = 0，j = array.length - 1。

class Solution {
public:
    inline void swap(int &a, int &b)
    {
        int temp = a;
        a = b;
        b = temp;
    }
    void sortColors(vector<int>& nums)
    {
        if(nums.size() <= 1)
        {
            return;
        }
        int i = 0;
        int j = nums.size() - 1;
        int cur = i;
        
        while(cur <= j)
        {
            if(nums[cur] == 0)
            {
                if(cur > i)
                {
                    swap(nums[i++], nums[cur]);
                }
                else
                {
                    //此时cur == i, A[i] == A[cur]
                    ++cur;
                    ++i;
                }
            }
            else if (nums[cur] == 2)
            {
                if(cur < j)
                {
                    swap(nums[j--], nums[cur]);
                }
                else
                {
                    return;
                }
            }
            else
            {
                ++cur;
            }
        }
    }
};