[小记]三色旗问题，可以转换为负数，0，正数划分问题

#include "stdafx.h"
/************************************************************************/
/*编程解决三色旗问题，一个字符型数组color，其中每个元素值为'R'，'W'，'B'三
/*者之一（代表红、蓝、白三色）编程实现把数组中元素重新排列：所有蓝色在前，
/*白色其后，红色最后。例如：原数组RWRRWBBRWRWR，重排后，BBBWWWRRRRR。
/************************************************************************/

#include <iostream>
using namespace std;

enum{
	BLUE='B',
	WHITE='W',
	RED='R'
};

inline void swap(char& f, char& e)
{
	char tmp=f;	f=e; e=tmp;
}

//三色旗问题标准算法
int OrderFlags(char flags[], int size)
{
	int bM, wM, rM;
	bM = wM = 0;
	rM= size - 1;

	int count=0;

	while(wM <= rM) //必须白色和红色标志重合
	{
		if(flags[wM] == WHITE)  //如果为白色，则白色标志后移
		{
			++ wM;
		}
		else if(flags[wM] == BLUE) //如果为蓝色，则跟第一个白色进行交换
		{
			swap(flags[wM],flags[bM]);
			++ count;
			++wM, ++bM; //白色和蓝色各增加一个
		}else{
			//如果是红色，则从尾部向前寻找第一个非红色的字符跟这个字符替换
			while(wM < rM && flags[rM] == RED) --rM; //前溯
			swap(flags[wM],flags[rM]); 
			++ count;

			--rM; //有一个红色归位了
		}
	}

	return count;
}



int _tmain(int argc, _TCHAR* argv[])
{
	char test[]="RWRRWBBRWRWR";
	int len=strlen(test);
	cout<<test<<" "<<len<<endl;
	cout<<OrderFlags(test,len)<<endl;
	cout<<test<<endl;
	return 0;
}