数组------荷兰国旗问题(颜色排序)

问题描述：
现在有红、白、蓝三色小球若干个，要求对其进行整理，使得同色小球相邻。
思路：
参照快速排序的过程，但使用三个指针first, current 和 end.其中first和current从数组头开始，end指向数组尾。以０表示红色，１表示白色，２表示蓝色。则有：
(1) 当*current = 0 时，swap(*current, *first) 并且first++、current++ ;
(2) 当*current = 1 时，current++;
(3) 当*current = 2 时，swap(*current, *end) 并且end–;

#include <iostream>
using namespace std;

void colorSort(int *arr, int first, int end);

int main()
{
    int array[] = {0, 2, 2, 1, 1, 0, 0, 1, 2, 1, 2};

    colorSort(array, 0, sizeof(array)/sizeof(int)-1);
    cout << "after sort: " << endl;
    for(int i = 0; i < sizeof(array)/sizeof(int); i++)
    {
        cout << array[i] << " ";
    }
    cout << endl;

    return 0;
}

void colorSort(int *arr, int first, int end)
{
    if(arr == NULL || first >= end)
        return ;
    int current = first;
    while(current < end)
    {
        if(arr[current] == 0)
        {
            swap(arr[current], arr[first]);
            current++;
            first++;
        }
        else if(arr[current] == 1)
        {
            current++;
        }
        else
        {
            swap(arr[current], arr[end]);
            end--;
        }
    }
}