荷兰国旗问题

【算法设计与分析基础4.2-9】

荷兰国旗问题要求对字符R、W和B构成的任意数组排序（红、白和蓝是荷兰国旗的颜色），使得所有R排在最前面，W随后，B在最后。为该问题设计一个线性效率的在位算法。

【算法】

假设数组的状态如下

A[0] ... A[r-1] A[r] ... A[w -1] A[w] ... A[b]  A[b+1] ... A[n -1]

     全部填R          全部填W             未处理部分                 全部填B

下一步处理A[w]，假如A[w]的值为R，则交换（A[r],A[w]），同时r <- r + 1, w <- w + 1。假如A[w]的值为W，只需要w <- w + 1。假如A[w]的值为B，交换（A[w],A[b]），同时b <- b -1，因为A[b]的值是未知的，我们需要对其进行判断，所以w不需要往前移。

伪代码：

DutchFlag(A[0...n-1])

r <- 0; w <- 0; b <- n-1

while w <= b do

       if A[w] ='R'

             swap(A[r], A[w]);  r <- r+1; w <- w+1;

       else if A[w] ='W'

             w <- w+1;

       else

            swap(A[w], A[b]);  b <- b-1 

【代码】

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

void swap(char *a, char *b)
{
	char tmp;
	tmp = *a;
	*a = *b;
	*b = tmp;
}

void DutchFlag(char a[], int n)
{
	int r, w, b;
	r = w = 0;
	b = n - 1;
	while(w <= b)
	{
		if(a[w] == 'R')
			swap(&a[w++], &a[r++]);
		else if(a[w] == 'W')
			w++;
		else
			swap(&a[w], &a[b--]);
	}
}

int main(void)
{
	char a[10] = "RWBRWBRWB";
	int n;
	n = strlen(a);
	DutchFlag(a, n);
	printf("%s\n", a);
	return 0;
}