075 Sort Colors 分类颜色

给定一个包含红色、白色和蓝色，且含有 n 个元素的数组，对它们进行排序，使得相同颜色的元素相邻，颜色顺序为红色、白色、蓝色。
此题中，我们使用整数 0, 1 和 2 分别表示红色，白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
进阶：
一个相当直观的解决方案是使用计数排序的 two-pass 算法。
首先，迭代计算出0，1 和 2 元素的个数，然后重写当前数组。
你能想出一个仅使用恒定空间的 one-pass 算法吗？
详见：https://leetcode.com/problems/sort-colors/description/

Java实现：

方法一：计数排序思想

class Solution {
    public void sortColors(int[] nums) {
        int i=0;
        int j=0;
        int k=0;
        for(int p=0;p<nums.length;++p){
            if(nums[p]==0){
                ++i;
            }else if(nums[p]==1){
                ++j;
            }else{
                ++k;
            }
        }
        for(int p=0;p<nums.length;++p){
            if(p<i){
                nums[p]=0;
            }else if(p>=i&&p<(i+j)){
                nums[p]=1;
            }else{
                nums[p]=2;
            }
        }
    }
}

方法二：

设置两个index，left记录第一个1的位置，left左边为0，right记录第一个非2的位置，right右边为2.
然后使用i从头到尾扫一遍，直到与right相遇。
i遇到0就换到左边去，遇到2就换到右边去，遇到1就跳过。
需要注意的是：由于left记录第一个1的位置，因此nums[left]与nums[i]交换后，nums[left]为0,nums[i]为1，因此i++；
而right记录第一个非2的位置，可能为0或1，因此nums[right]与nums[i]交换后，nums[right]为2,nums[i]为0或1，i不能前进，要后续判断。
由此该数组分为4段：[0,left)-->0; [left,i)-->1; [i,right]-->乱序; (right,n-1]-->2

class Solution {
    public void sortColors(int[] nums) {
        int left=0;
        int right=nums.length-1;
        int i=0;
        while(i<=right){
            if(nums[i]==0){
                swap(nums,i,left);
                ++left;
                ++i;
            }else if(nums[i]==1){
                ++i;
            }else{
                swap(nums,i,right);
                --right;
            }
        }
    }
    private void swap(int[] nums,int i,int j){
        int tmp=nums[i];
        nums[i]=nums[j];
        nums[j]=tmp;
    }
}

参考：https://www.cnblogs.com/ganganloveu/p/3703746.html