榆木脑壳练算法之三色旗问题

    今天闲来无事，写了一个关于三色旗帜的问题的代码，参见书上的思路，也写了自己的一些理解，记录与此，权当参考。
    问题：假设有一条绳子，上面有红、白、蓝三种颜色的旗子，起初绳子上的旗子颜色并没有顺序，您希望将之分类，并排列为蓝、白、红的顺序，要如何移动次数才会最少，注意您只能在绳子上进行这个动作，而且一次只能调换两个旗子

    解法：

// ThreeColorFlags.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <process.h>

enum FLAG{
	COLOR_RED = 0,
	COLOR_WHITE,
	COLOR_BLUE
};

void swapFlag(FLAG &flag1, FLAG &flag2)
{
	FLAG temp;
	
	temp = flag1;
	flag1 = flag2;
	flag2 = temp;
}

/************************************************************************/
/*解题思路：说简单点,你在绳子上遍历每个绳子的颜色，遇见蓝色的向前方，白色
**白色的方中间，红色的向后放；下面具就是分别设立每个颜色旗帜的哨兵来表示处
**理过的颜色的旗帜的位置，但是遍历的时候，要以你所排列顺序的中间颜色为基准
**（这里为白色），因为这样，既表示遍历到的位置，其他两个颜色会滞后于中间颜
色的变化，进行移动的时候，才能正确的交换顺序
*/
/************************************************************************/
void moveStrategy(FLAG* flags, const int flagNum)
{
	int blueIndex = 0;
	int whiteIndex = 0;
	int redIndex = flagNum - 1;

	while(whiteIndex <= redIndex)
	{
		if(flags[whiteIndex] == COLOR_WHITE)
		{
			//当前的旗帜为白色，将白色旗帜的哨兵后移
			whiteIndex++;
		}
		else if(flags[whiteIndex] == COLOR_BLUE)
		{
			//当前的旗帜为蓝色，交换蓝色哨兵和白色哨兵所指代的旗帜，并将蓝色和白色哨兵都后移
			swapFlag(flags[whiteIndex], flags[blueIndex]);
			whiteIndex++;
			blueIndex++;
		}
		else
		{
			//当前旗帜为红色，则将当前旗帜和红色的旗帜的哨兵交换，并将红色旗帜哨兵前移，此时交换到
			//前面的旗帜不知道颜色，所以其他颜色的哨兵不动
			swapFlag(flags[whiteIndex], flags[redIndex]);
			redIndex--;
		}
	}
}

void printFlags(FLAG *flags, const int flagNum)
{
	for(int index = 0; index < flagNum; index++)
	{
		if(flags[index] == COLOR_WHITE)
		{
			printf("w ");
		}
		else if (flags[index] == COLOR_BLUE)
		{
			printf("b ");
		}
		else
		{
			printf("r ");
		}
	}
}

//把一条绳子上的旗帜排列为蓝，白，红的顺序
int _tmain(int argc, _TCHAR* argv[])
{
	FLAG totalFlags[] = {COLOR_RED, COLOR_WHITE, COLOR_BLUE, COLOR_WHITE, COLOR_WHITE,
		COLOR_BLUE, COLOR_RED, COLOR_BLUE, COLOR_WHITE, COLOR_RED};
	int numFlag = sizeof(totalFlags) / sizeof(FLAG);

	moveStrategy(totalFlags, numFlag);
	printFlags(totalFlags, numFlag);

	system("pause");
	return 0;
}