lintcode:颜色分类

颜色分类

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

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

样例

给你数组 [1, 0, 1, 2], 需要将该数组原地排序为 [0, 1, 1, 2]

解题

暴力：两次快排 、扫描数组，记录0 1 2 的个数，更新数组

下面是一次快排 + 线性查找

对于数组 A，两个指针 low high

low指向 0的都跳过，结束的时候指向： 1或 2

high指向2的都跳过，结束的时候指向 ：0 或 1

当是(1 ,0)时候，交换 low++ 使得0归位

当是(2,0)时候，交换high-- 使得 2归位

当是(2,1)时候，交换high-- 是的2归位

当时(1,1)时候，就比较复杂了

先左找，找到第一个0 和左1互换 low++

找不到0的时候从右1开始右找，找到第一个2的时候，和右1互换 high--

当左找不到2的时候，说明low  high之间全是 1 结束

时间复杂度O(N)

class Solution {
    /**
     * @param nums: A list of integer which is 0, 1 or 2
     * @return: nothing
     */
    public void sortColors(int[] A) {
        // write your code here
        if(A == null)
            return;
        int len = A.length;
        int low = 0;
        int high = len - 1;
        while(low < high){
            while(low <= high && A[low] == 0) low++;
            while(low <= high && A[high] ==2) high--;
            if(low < high ){
                if(A[low] == 1 && A[high] ==0){
                    A[low] = 0;
                    A[high] = 1;
                    low++;
                }else if(A[low]==2 && A[high] ==0){
                    A[low] =0;
                    A[high] =2;
                    low++;
                    high--;
                }else if(A[low]==2 && A[high] ==1){
                    A[low] =1;
                    A[high] =2;
                    high--;
                }else if(A[low] == A[high] && A[low] == 1){
                    // 左开始找0
                    int tmp0 = low + 1;
                    while(tmp0 <= high && A[tmp0] != 0) tmp0++;
                    // 找到了 0 和左1交换
                    if(tmp0 <= high && A[tmp0]==0){
                        A[low] = 0;
                        A[tmp0] = 1;
                        low++;
                    }else { //没找到 0
                        // 右开始找2
                        int tmp2 = high - 1;
                        while(tmp2 >= low && A[tmp2] != 2) tmp2--;
                        // 找到 2 和右 1 交换
                        if(tmp2 >=low && A[tmp2] == 2){
                            A[high] = 2;
                            A[tmp2] = 1;
                            high--;
                        }else{//没找到 0 也没找到 2 说明全是 1 结束
                            return;
                        }
                    }
                }
            }
        }
    }
}