荷兰国旗问题

// FlagArrange.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"

#include <stdio.h>

/*
由红白蓝三种颜色的球组成的序列，编写算法，使得该序列排列次序为红球，白球，蓝球，且时间复杂度为O(n);

思路：假设红白蓝三种颜色依次用数字1,2,3来表示，则问题变为：12223311123122221111113333，将这些随机序列变为111111122222333
*/
void swap(int &a, int &b)
{
    int temp = a;
    a = b;
    b = temp;

    return;
}

int len(int list[])
{
    int length = 0;
    
    while(list[length] != 0)
        length++;

    return length;
}

void display(int list[])
{
    printf("List: ");
    
    for(int i = 0; i < len(list); i++)
    {
        printf("%d",list[i]);
    }
    
    printf("\n\n");

    return;
}

void arrange(int list[])
{
    int i = 0;      //list[i]之前为1
    int j = 0;      //list[j]为当前需要处理的颜色
    int k = len(list) -1; //list[k]之后的为3

    while(j <= k)
    {
        switch(list[j])
        {
        case 1:
            swap(list[i],list[j]);
            i++;
            j++;
            break;
        case 2:
            j++;
            break;
        case 3:
            swap(list[k],list[j]);
            k--;
            break;
        }
    }

    return;
}

int _tmain(int argc, _TCHAR* argv[])
{
    int list[] = {1,2,3,1,2,1,2,3,1,1,1,2,2,2,2,1,1,3,3,2,1,0};
    display(list);
    arrange(list);
    display(list);

    getchar();
    return 0;
}