[leetcode]Sort Colors

题目地址： https://oj.leetcode.com/problems/sort-colors/

题目大致含义：一个数组有3个数字：0,1,2.将这个数组排序。0在最前面，2在最后面。  如果可以，尽量只扫描一遍数组。

method:

第一种想法：

扫描一遍数组，记录0,1，2的个数。第二遍分别写入对应个数的0,1,2.

代码如下：

class Solution {
public:
    void sortColors(int A[], int n) {
        int red=0, white=0, blue=0;
        for(int i=0; i<n; i++)
        {
            if(A[i] == 0) red++;
            if(A[i] == 1) white++;
            if(A[i] == 2) blue++;
        }
        int pos = 0;
        while(red-- >0)
        {
            A[pos++] = 0;
        }
        while(white-- >0)
        {
            A[pos++] = 1;
        }
        while(blue-- >0)
        {
            A[pos++] = 2;
        }   
    }
};

这个方法思路比较简单，代码也比较清晰。

如果只扫描一次数组的话。则需要对应额外的指针。

代码如下：

class Solution {
public:
    void sortColors(int A[], int n) {
        int i=-1, j=-1, k=-1;
        for(int p = 0; p < A.length; p++)
        {
            if(A[p] == 0)
            {
                A[++k]=2;
                A[++j]=1;
                A[++i]=0;
            }
            else if (A[p] == 1)
            {
                A[++k]=2;
                A[++j]=1;
    
            }
            else if (A[p] == 2)
            {
                A[++k]=2;
            }
        }
        
    }
};

这个方法用i,j,k分别指向当前0,1,2的位置。但是这个方法存在的一个问题是：额外的写比较多。

下面的这个方法类似上面的方法，但是减少了额外的写。

代码：

class Solution {
public:
    void sortColors(int A[], int n) {
        int j = 0, k = n;
        for(int i=0; i<k; i++)
        {
            if(A[i] == 0)
                swap(A[j++], A[i]);
            else if(A[i] == 2)
                swap(A[--k], A[i--]);
        }
    }
};