题目描述 现在有一个包含n个物体的数组,其中物体颜色为颜色为红色、白色或蓝色,请对这个数组进行排序,让相同颜色的物体相邻,颜色的顺序为红色,白色,蓝色。 我们用0,1,2分别代表颜色红,白,蓝 注意: 本题要求你不能使用排序库函数 扩展: 一个非常直接的解法是两步的计数排序的算法 首先:遍历一遍数组,记录0,1,2的数量,然后重写这个数组,先将0写入,再将1写入,再将2写入 你能给出一个只用一步,并且能在常数级空间复杂度解决这个问题的算法吗?

在用常数N的复杂度解决问题
首先,判断数组长度为1和2的情况,定义左右标记left和right
其次,用fori遍历数组,等于0和left交换,等于2和right交换,否则i才可以往后移动
最后注意的细节是循环结束条件应该是i<=right,i等于left并且此时为0时i可以直接往后移动

实现的代码:

public void swap(int[] A,int left,int right){
        int temp = A[left];
        A[left] = A[right];
        A[right] = temp;
    }
    public void sortColors(int[] A) {
        if (A.length==1)return;
        if (A.length==2)if (A[0]>A[1])swap(A,0,1);
        int left=0,right=A.length-1;
        for (int i = 0; i < A.length; ) {
            if(i<=right) {
                if (A[i] == 0&&i!=left) {
                    swap(A, i, left);
                    left++;
                } else if (A[i] == 2) {
                    swap(A, i, right);
                    right--;
                } else if (left==right){break;}
                else i++;
            }
            else
                break;
        }
    }