一个练手的程序

typedef enum color_e
{
    RED,
    WHITE,
    BLUE,
} color_t;
 
int __partition(color_t arr[], int _start, int _end, color_t c)
{
    int start = _start;
    int end = _end;
 
    while(start < end)
    {
        while(start <= end && arr[start] == c) start++;
        while(start <= end && arr[end] != c) end--;
        if(start<end)
        {
            color_t tmp = arr[start];
            arr[start] = arr[end];
            arr[end] = tmp;
        }
    }
    return start;
}
 
void partition(color_t arr[], int n)
{
    __partition(arr, __partition(arr, 0, n - 1, WHITE), n - 1, BLUE);
}
 
bool check(color_t arr[], int n)
{
    int i = 0;
    while(i < n && arr[i] == WHITE) i++;
    while(i < n && arr[i] == BLUE) i++;
    while(i < n && arr[i] == RED) i++;
 
    assert(i == n);
    return i == n;
}
 
void init(color_t arr[], int n)
{
    int i = 0;
    color_t tmp_arr[3] = {WHITE, BLUE, RED};
    while(i < n)
    {
        arr[i] = tmp_arr[rand()%3];
        i++;
    }
}
 
int main()
{
    srand(time(0));
    int i = 1;
 
    for( ; i<100000; i++)
    {
        color_t * arr = new color_t[i];
        init(arr, i);
        partition(arr, i);
        if(check(arr, i))
        {
            printf("===================%d\n", i);
        }
        delete []arr;
    }
}