荷兰国旗问题

荷兰国旗问题：

现有红，白，蓝三个不同颜色的小球，乱序排列在一起，重新排列这些小球，使得红白蓝三色的同颜色的球在一起。

问题分析：

问题转换为：给定数组A[0,1,...,N-1]，元素只能取0，1，2三个值，设计算法使得数组重新排列成“000...111..222”的形式。

可以使用三个游标，begin=0，cur=0，end=N-1。

程序实现：

/***************************************
FileName HollandFlag.cpp
Author : godfrey
CreatedTime : 2016/5/4
****************************************/
#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
#include <stdio.h>
#include <stdlib.h>

using namespace std;

void HollandFlag(int* A,int size){
    int begin = 0;
    int end = size - 1;
    int cur = 0;
    while(cur <= end){
        if(A[cur] == 2){
            swap(A[cur],A[end]);
            end --;
        }
        else if(A[cur] == 1){
            cur ++;
        }
        else if(A[cur] == 0){
            swap(A[cur],A[begin]);//这里简化了步骤，提醒：cur与begin是否相同，分情况考虑
            begin ++;
            cur ++;
        }
    }
}
int main()
{
    int a[] = {0,1,2,0,0,2,1,2,1,2,2,1,1,0};
    int size = sizeof(a)/sizeof(int);
    for(int i=0;i<size;i++){
        cout<<a[i]<<" ";
    }
    cout<<endl;
    HollandFlag(a,size);
    cout<<"-----------After adjusting ------------"<<endl;
    for(int i=0;i<size;i++){
        cout<<a[i]<<" ";
    }
    cout<<endl;
    return 0;
}

运行结果：

转载请注明出处：

C++博客园：godfrey_88

http://www.cnblogs.com/gaobaoru-articles/