现有红白蓝三个不同颜色的小球,乱序排列在一起,请重新排列这些小球,使得红白蓝三色的同颜色的球在一起。这个问题之所以叫荷兰国旗,是因为我们可以将红白蓝三色小球想象成条状物,有序排列后正好组成荷兰国旗。
数字0:红球;数字1:白球;数字2:篮球
思路:使用三个指针,一个前指针begin,一个中指针current,一个后指针end。Current指针遍历整个数组序列。
我们先来回顾一下快速排序:
快速排序时基于分治模式处理的,对一个典型子数组A[p...r]排序的分治过程为三个步骤:
1、分解:A[p..r]被划分为俩个(可能空)的子数组A[p ..q-1]和A[q+1 ..r],使得A[p ..q-1] <= A[q] <= A[q+1 ..r] ;
2、解决:通过递归调用快速排序,对子数组A[p ..q-1]和A[q+1 ..r]排序;
3、合并。
也就是说,快速排序的主要思想便是依托于一个partition分治过程,每一趟排序的过程中,选取的主元都会把整个数组排列成一大一小的序列,继而递归排序完整个数组。
然后,我们再回来看看本题:
1、current指针指向元素0时,与begin指针所指向的元素交换,而后current++,begin++;2、current指针指向元素1时,current指1不做交换,然后current++;
3、current指针指向元素2时,与end指针所指向的元素交换,而后,如果current指针指向0时,current指针此刻不动,按照第一步执行;若current指针指向1,按第二步执行,且此时end--。在操作上可以直接使current指针不动,end--即可。
//荷兰国旗
#include <iostream>
using namespace std;
//Holland flag
void OddEvenSort(int *pData, unsigned int length)
{
if (pData == NULL || length == 0)
{
return;
}
int *begin = pData;
int *current = pData;
int *end = pData + length - 1;
while (current <= end)
{
if (*current==0)
{
swap(*begin, *current);
begin++;
current++;
}
if (*current == 1)
{
current++;
}
if (*current == 2)
{
swap(*current,*end);
end--;
}
}
}
int main()
{
int a[10] = { 0,1,2,1,1,2,0,2,1,0 };
OddEvenSort(a, 10);
for (int i = 0; i < 10; i++)
{
cout << a[i] << " ";
}
cout << endl;
return 0;
}