颜色分类（LintCode）

颜色分类

给定一个包含红，白，蓝且长度为n的数组，将数组元素进行分类使相同颜色的元素相邻，并按照红、白、蓝的顺序进行排序。

我们可以使用整数0，1和2分别代表红，白，蓝。

样例

 

注意

不能使用代码库中的排序函数来解决这个问题

说明

一个相当直接的解决方案是使用计数排序扫描2遍的算法。

首先，迭代数组计算0,1,2出现的次数，然后依次用0,1,2出现的次数去覆盖数组。

你否能想出一个仅使用常数级额外空间复杂度且只扫描遍历一遍数组的算法？

 

代码写的乱糟糟的，难得写了注释。

想法就是把0交换至左边，2交换至右边。

class Solution {
    //总耗时: 14896 ms
    
    /**
     * @param nums: A list of integer which is 0, 1 or 2 
     * @return: nothing
     */
    public void sortColors(int[] nums) {
        int low = 0;
        int high = nums.length-1;
        //先找到最左的非0和最右的非2
        while(nums[low] == 0 && low < high) low++;
        while(nums[high] == 2 && low < high) high--;

        while(nums[low] == 2 || nums[high] == 0 && low < high) {
            //如果最左非0是2或者最右非2是0，则将其换至右边或左边
            if(nums[low] == 2) {
                if(nums[high] == 0) {
                    nums[low++] = 0;
                    nums[high--] = 2;
                }else {
                    nums[high--] = 2;
                    nums[low] = 1;
                }
            }else {
                if(nums[high] == 0) {
                    nums[low++] = 0;
                    nums[high] = 1;
                }
            }
            //找到最左的非0和最右的非2
            while(nums[low] == 0 && low < high) low++;
            while(nums[high] == 2 && low < high) high--;
        }
        
        for(int i=low+1;i<high && low < high;i++) {
            //此时最左非0和最右非2都是1，用i找到0或2，交换到左端或右端
            if(nums[i] == 0) {
                nums[low++] = 0;
                nums[i] = 1;
            }else {
                if(nums[i] == 2) {
                    nums[high--] = 2;
                    nums[i] = 1;
                }
            }
            
            //找到最左的非0和最右的非2
            while(nums[low] == 0 && low < high) low++;
            while(nums[high] == 2 && low < high) high--;
            
            while(nums[low] == 2 || nums[high] == 0 && low < high) {
                //如果最左非0是2或者最右非2是0，则将其换至右边或左边
                if(nums[low] == 2) {
                    if(nums[high] == 0) {
                        nums[low++] = 0;
                        nums[high--] = 2;
                    }else {
                        nums[high--] = 2;
                        nums[low] = 1;
                    }
                }else {
                    if(nums[high] == 0) {
                        nums[low++] = 0;
                        nums[high] = 1;
                    }
                }
                //找到最左的非0和最右的非2
                while(nums[low] == 0 && low < high) low++;
                while(nums[high] == 2 && low < high) high--;
            }
            //若此时的low>=i，显然要重新设置i
            if(i <= low) i = low+1;
        }
        //当上面的循环结束时，分类就已完成，只遍历了一次nums，辅助空间为常数
    }
}