荷兰红旗问题

设有一个仅由红、白、蓝三种颜色的条块组成的条块序列，请编写一个时间复杂度为O（n）的算法，使得这些条块按红、白、蓝的顺序排好，即排成荷兰国旗图案。
 1 #include <stdio.h>
#include <algorithm>
using namespace std;

typedef enum
{
    RED,
    WHITE,
    BLUE
} color; //定义荷兰国旗颜色枚举数组

void flag_arrange(color sz[], int len)
{
    // i以前全为红色，k以后全为蓝色，i和k中间
    int i = 0, j = i, k = len - 1;
    while (j <= k)
    {
        // 如果为红色，则和i交换
        if (sz[j] == RED)
            swap(sz[i++], sz[j++]);
        else if (sz[j] == BLUE) // 如果为蓝色，则和k交换
            swap(sz[k--], sz[j]); //这里没有j++以防交换后仍为蓝色的情况
        else
            j++;
    }
}
int main()
{
    // 定义一个荷兰国旗数组
    color a[] = {RED, WHITE, WHITE, BLUE, BLUE, RED, WHITE, BLUE, RED, RED, BLUE, WHITE};
    // 执行排序
    flag_arrange(a, 12);
    // 打印输出排序结果
    for (int i = 0; i < 12; i++)
        printf("%d%c", a[i], i == 11 ? '\n' : ' ');
    return 0;
}