LeetCode #75 - Sort Colors

题目描述：

Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note:
You are not suppose to use the library's sort function for this problem.

click to show follow up.

Follow up:
A rather straight forward solution is a two-pass algorithm using counting sort.
First, iterate the array counting number of 0's, 1's, and 2's, then overwrite array with total number of 0's, then 1's and followed by 2's.

Could you come up with an one-pass algorithm using only constant space?

只需要一次遍历就能把0,1,2按先后顺序全部划分到数组的相邻位置，即从数组两端开始，用两个指针p,q指向已经划分好了的0和2在数组两端的区域外的第一个元素，然后用指针i遍历数组，如果当前元素为0，那么将它与p指向的元素交换，同时修改指针；如果当前元素为2，那么将它与q指向的元素交换，同时修改指针。当0和2的区域在数组两端的区域都划分好后，剩下的中间部分就是1的区域。

特别需要注意的是，在nums[i]=0时，存在有些情况，比如初始的时候i=p，此时为了避免p>i，必须令p++、i++，而如果i>p，那么p指向的元素只可能是1（p指向0区域的后一位，指针i之前的0全部划分至0区域中，指针i之前的2都已经被交换至数组后端的2区域中），在这种情况下令i++也没有问题。所以对nums[i]=0，必须直接令i++，也就是说从p指针交换过来的元素不需要判断；但是从q指针交换过来的元素需要判断，它可能是0或1或2，此时i保持不变，等待下一次循环判断。

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int p=0;
        int q=nums.size()-1;
        int i=0;
        while(i<=q)
        {
            if(nums[i]==0)
            {
                nums[i]=nums[p];
                nums[p]=0;
                p++;
                i++;
            }
            else if(nums[i]==2)
            {
                nums[i]=nums[q];
                nums[q]=2;
                q--;
            }
            else if(nums[i]==1) i++;
        }
    }
};