荷兰国旗问题

描述
 
荷兰国旗有三横条块构成，自上到下的三条块颜色依次为红、白、蓝。现有若干由红、白、蓝三种颜色的条块序列，
要将它们重新排列使所有相同颜色的条块在一起。
本问题要求将所有红色的条块放最左边、所有白色的条块放中间、所有蓝色的条块放最右边。
 
输入
第1行是一个正整数n（n<100），表示有n组测试数据。接下来有n行，每行有若干个由R，W，B三种字符构成的字符串序列，
其中R，W和B分别表示红、白、蓝三种颜色的条块,每行最多有1000个字符。
输出
对输入中每行上由R，W，B三种字符构成的字符串序列，将它们重新排列使所有相同颜色的条块在一起，满足前述要求。
样例输入
3
BBRRWBWRRR
RRRWWRWRB
RBRW 
样例输出
RRRRRWWBBB
RRRRRWWWB
RRWB 

#include <iostream> 
#include <stdlib.h> 
using namespace std;
#define N 10

void swap (int &var1, int &var2)
{      
	int temp = var1;    
	var1 = var2;      
	var2 = temp; 
}

void shuffle(int *array) 
{      
	int current = 0;     
	int end = N-1;      
	int begin = 0;      
	while( current<=end )     
	{           
		if( array[current] ==0 )          
		{               
			swap(array[current],array[begin]);
			current++;                
			begin++; 
		}           
		else if( array[current] == 1 )
		{                
			current++;           
		}           
		else{//When array[current] =2
			swap(array[current],array[end]);
			end--;
		}      
	} 
}
	
int main(int argc, char *argv[]) 
{     
	int a[N]; 
	int i; 
	for( i=0 ; i<N; i++ )
	{ 
		a[i] = rand()%3; 
		cout << a[i] << " ";
	} 
	cout << endl;
	shuffle(a);
	for( int i=0 ; i<N ; i++ )
	{     
		cout << a[i] << " ";  
    } 
	cout << endl;   
	return 0; 
}