Sort Colors

其实这个Follow up说的也很好哇，计数排序

A rather straight forward solution is a two-pass algorithm using counting sort.
First, iterate the array counting number of 0's, 1's, and 2's, then overwrite array with total number of 0's, then 1's and followed by 2's.

    void sortColors(int A[], int n) {
        if(A==NULL||n<=0)
            return;
        int count=3;
        int C[count];
        int i,j;
        for(i=0;i<count;++i)
            C[i]=0;
        for(i=0;i<n;++i){
            j=A[i];
            C[j]++;
        }
        j=n-1;
        for(i=count-1;i>=0;--i){
            while(C[i]){
                A[j]=i;
                C[i]--;
                j--;//这个别忘了
            }
        }
    }

 AC

如果只遍历一遍的话，我觉得快排挺合适的，但是万一数组中没有1呢，是不是要先扫描一下，看看是不是有1，那不能，如果那样的话，也要扫两遍的。那怎么办呢?

管他三七二十一呢，就把数组中的最后一个存到一个变量中先，然后把它设为1，然后等所有的都排好后，就把变量中的那个放到那个位置，

如果是 0,0,0，*，1,1,1,2   那么原来数组中是0或1都可以，可是如果是2的话就不行了

如果是 0,0,0，1,1,1,*，2   那么原来数组中是2或1都可以，可是如果是0的话就不行了

如果是 0,0,0，1,1,1,*，1，1，1，1，2   那么原来数组中是1可以，可是如果是0或者2的话就不行了

可见，自己对快排的过程是不清楚的，找时间也得在仔细看看快排哇？？？？？？？？？？？？？？？？？？？？？？？？？

在 http://tech-wonderland.net/blog/leetcode-sort-colors.html 上看到了一个好方法

By keeping three pointers, i (=0),  j (= n -1), k, i points to the front part of the array while j points to the back end of the array, use k to scan through the whole array, when encountering 0, swap A[i] and A[k], when encountering 2, swap A[j] and A[k] and move the pointers i, j, k correspondingly, after finishing one pass scan, all the elements before i are expected to be 0, all the elements after j are expected to be 2, all the elements between i and j are 1.

    void swap(int &a,int &b){
        int tmp;
        tmp=a;
        a=b;
        b=tmp;
    }
    void sortColors(int A[], int n) {
        if(A==NULL||n<=0)
            return;
        int i,j,k;
        i=0;
        j=n-1;
        k=0;
        while(k<=j){
            if(A[k]==0){
                swap(A[i],A[k]);
                i++;
            }
            if(A[k]==2){
                swap(A[k],A[j]);
                j--;
            }
            k++;
        }
    }

Status: Wrong Answer

Input:[1,2,0]

Output:[1,0,2]

Expected:[0,1,2]

http://tech-wonderland.net/blog/leetcode-sort-colors.html中还说道：

Tips and Divergent thinking:

Be careful when moving the pointers: when to just move j e.g j–, when to just move i e.g. i++, when to just move k, k++, when to move both i and k, or move both j and k.

如果把23行改为

if(A[k]!=0&&A[k]!=2)
　　k++;

当A[]={0}时，就是死循环了。

    void swap(int &a,int &b){
        int tmp;
        tmp=a;
        a=b;
        b=tmp;
    }
    void sortColors(int A[], int n) {
        if(A==NULL||n<=0)
            return;
        int i,j,k;
        i=0;
        j=n-1;
        k=0;
        while(k<=j)
        {
            if(A[k]==2)
            {
                swap(A[k],A[j]);
                j--;
            }
            if(A[k]==0)
            {
                swap(A[k],A[i]);
                i++;
                k++;
            }
            if(A[k]!=0&&A[k]!=2)
                k++;
        }
    }

AC

明哥说，等于0的时候可以k++，等于2时不能k++

不理解