排列颜色

题目描述

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

class Solution {
public:
    void sortColors(int A[], int n) {
        int front = 0, tail = n-1;
        int i = 0;
        while(i <= tail){
            // 把0放到前面
            if(A[i] == 0){
                swap(A[front], A[i]);
                i++;
                front++;
            }
            // 把2放到后面
            else if(A[i] == 2){
                swap(A[tail], A[i]);
                tail--;
            }
            else{
                i++;
            }
        }
    }
};

问题原因

• 第一次错是直接使用快排的思想，大的放右边，小的放左边，但没有考虑到，pivot为0时，会出现 021这样的错误情况。
• 第二次错误是把大的移到右边，交换来的大小是无法确定的，所以此时 i 的位置是不能动的