算法 --- leetcode之sort-colors

题目描述：

现在有一个包含n个物体的数组，其中物体颜色为颜色为红色、白色或蓝色，请对这个数组进行排序，让相同颜色的物体相邻，颜色的顺序为红色，白色，蓝色。

我们用0,1,2分别代表颜色红，白，蓝

注意：

本题要求你不能使用排序库函数

扩展：

一个非常直接的解法是两步的计数排序的算法

首先：遍历一遍数组，记录0,1,2的数量，然后重写这个数组，先将0写入，再将1写入，再将2写入

你能给出一个只用一步，并且能在常数级空间复杂度解决这个问题的算法吗？

Java源码实现：

public class Solution {
    public void sortColors(int[] A) {
        int length = A.length;
        int lt = 0;
        int rt = length - 1;
        for (int i = 0; i <= rt; i++) {
            // 当A[i]为0时A[i]与A[lt]进行交换，此时A[i]处的0已经去到了它该在的位置上
            if (A[i] == 0) { 
              int temp = A[lt];
              A[lt] = A[i];
              A[i] = temp;
              lt++;
            }
            if (A[i] == 2) {
              int temp = A[rt];
              A[rt] = A[i];
              A[i] = temp;
              rt--;
              // i进行减1操作因为此时A[rt]的值不确定，不知道其位置是否正确。
              // 但是后续会进行i++操作，这里进行i--只是为了保持当前位置不变
              i--; 
            }
        }
    }
}