数组的应用（二）----荷兰国旗问题的解决

众所周知，荷兰国旗由红色，白色和蓝色3种颜色组成。现在假设有很多这3种颜色的线被存放在一个数组里面，要求每次操作只能进行一次交换，对数组进行一遍扫描后，3种颜色自然分开，颜色顺序应为红，白，蓝。要求时间复杂度O(n)，使移动次数最少。

解决思路：荷兰国旗问题其实就是一个排序问题，可以将红，白，蓝3种颜色分别用数字0,1,2表示，用一个数组来存储他们。将相同颜色线条归为一类就相等于将数组中的数值按大小进行排序，只不过数组里存储的只有3种数值而已。解题的基本策略是遍历两个颜色区域，如果颜色条不属于所在区域，则交换一个属于该区域的颜色条。这样，每一次都是有必要的交换，从而实现最少交换次数。程序实现代码如下：（添加了一点关键注释，助于分析代码）

 

#include<iostream>
using namespace std;
const int N=100;
int flag[N]; //国旗颜色条数组
int pre[N];//记录该白条的前白条位置
int split1;//区域分隔1
int split2;//区域分隔2
int blue_red;//红条在蓝色区域中的标示
int white_red;//红条在白色区域中的标记
int counts=0;
void out()
{
 for(int i=0;i<N;i++)
  cout<<flag[i];
 cout<<endl;
}
void swap(int& x,int& y)
{
 int temp=x;
 x=y;
 y=temp;
 counts++;
}
void work()
{
 for(int i=0;i<split1;i++)//红色区域，交换非红条
 {
  if(flag[i]!=0)//如果不属于该区域
  {
   if(blue_red>=split2)//如果蓝区域中还有红条，交换
   {
    swap(flag[i],flag[blue_red]);
    blue_red=pre[blue_red];
   }
   else//否则，和白区域中的红条交换
   {
    swap(flag[i],flag[white_red]);
    white_red=pre[white_red];
   }
  }
 }
 int b=N-1;//白，蓝区域
 for(i=split1;i<split2;i++)
 {
  if(flag[i]!=1)
  {
   while(flag[b]==2)
    b--;
   swap(flag[i],flag[b]);
   b--;
  }
 }
  
 
}
void init()
{
 int red_num=0;//统计红色条数
 int white_num=0;//统计白色条数
 int preI=-1;
 for(int i=0;i<N;i++)
 {
  flag[i]=rand()%3;
  if(flag[i]==0)
  {
   red_num++;
   pre[i]=preI;
   preI=i;
  }
  else
   if(flag[i]==1)
   {
    white_num++;
   }
 }

//将国旗分成3个颜色区域：0~split1-1(红),split1~split2-1(白),split2~N-1(蓝)
 split1=red_num;
 split2=red_num+white_num;
 blue_red=preI;
 i=split2-1;
 while(flag[i]!=0) i--;
 white_red=i;
}
int main()
{
 init();
 cout<<"原始:"<<endl;
 out();
 work();
 cout<<"移动"<<counts<<"次："<<endl;
 out();
 return 0;
}