说明
三色旗的问题最早由E.W.Dijkstra所提出,他所使用的用语为Dutch Nation Flag(Dijkstra为荷兰人),而多数的作者则使用Three-Color Flag来称之。
假设有一条绳子,上面有红、白、蓝三种颜色的旗子,起初绳子上的旗子颜色并没有顺序,您希望将之分类,并排列为蓝、白、红的顺序,要如何移动次数才会最少,注意您只能在绳子上进行这个动作,而且一次只能调换两个旗子。
解法
在一条绳子上移动,在程式中也就意味只能使用一个阵列,而不使用其它的阵列来作辅助,问题的解法很简单,您可以自己想像一下在移动旗子,从绳子开头进行,遇到蓝色往前移,遇到白色留在中间,遇到红色往后移,如下所示:
只是要让移动次数最少的话,就要有些技巧:
如果图中W所在的位置为白色,则W+1,表示未处理的部份移至至白色群组。
如果W部份为蓝色,则B与W的元素对调,而B与W必须各+1,表示两个群组都多了一个元素。
如果W所在的位置是红色,则将W与R交换,但R要减1,表示未处理的部份减1。
注意B、W、R并不是三色旗的个数,它们只是一个移动的指标;什幺时候移动结束呢?一开始时未处理的R指标会是等于旗子的总数,当R的索引数减至少于W的索引数时,表示接下来的旗子就都是红色了,此时就可以结束移动,如下所示:
从绳子的开头(最左边)开始,遇到蓝色的往前移,遇到白色的不动,遇到红色的往后移。
注:我们这里说的前移指往左,后移指往右。
再来看具体的代码实现,先看下图:
我们可以定义三个指针 a、b、c 。从上图可以看出,a和b指向第一个元素,c指向最后一个元素。a的作用是确保蓝色的旗子移到左边,而c的作用是确保红色的旗子移到右边。b则用于顺序访问各个旗子。
思路如下:查看b所指向的旗子的颜色,如果是白色的,则不移动任何旗子(因为要求白色的旗子在中间,所以不去动它),将b往后移一位。
如果是红色的,此时将b所指向的旗子和c所指向的旗子交换位置,同时c往前移一位。(交换后c指向的已经是红色的旗子了,所以可以往前移一位了)。
如果是蓝色的,此时将b所指向的旗子和a所指向的旗子交换位置,同时a往后移一位。(交换后a指向的已经是蓝色的旗子了,所以可以往后移一位了)。这里要注意的是b也要后移一位。(这里我也没有弄得很清楚。大概是b不能在a的后面)
那么什么时候停止交换呢?显然是c都跑到b的后面去的时候,即b>c
#include <stdio.h>
#include<stdlib.h>
#include<string.h>
#define BLUE 'b' #define WHITE 'w' #define RED 'r' #define SWAP(x, y) { char temp; temp = color[x]; color[x] = color[y]; color[y] = temp; } int main() { char color[] = {'r', 'w', 'b', 'w', 'w', 'b', 'r', 'b', 'w', 'r', '\0'}; int wFlag = 0; int bFlag = 0; int rFlag = strlen(color) - 1; int i; for(i = 0; i < strlen(color); i++) printf("%c ", color[i]); printf("\n"); while(wFlag <= rFlag) { if(color[wFlag] == WHITE) wFlag++; else if(color[wFlag] == BLUE) { SWAP(bFlag, wFlag); bFlag++; wFlag++; } else { while(wFlag < rFlag && color[rFlag] == RED) rFlag--; SWAP(rFlag, wFlag); rFlag--; } } for(i = 0; i < strlen(color); i++) printf("%c ", color[i]); printf("\n"); return 0; }
java
import java.util.Scanner; public class TricolorChess { public static void main(String[] args){ System.out.println("Enter TricolorChess-like'rbwwrrrbbw': "); Scanner scanner = new Scanner(System.in); String s = scanner.next(); TricolorChess tcf = new TricolorChess(); s = tcf.ChessSort(s.toLowerCase().toCharArray()); System.out.println("out char:"+s); scanner.close(); } /** * 交换 * @param flags * @param x * @param y */ public void Swap(char[]flags, int x, int y){ char temp = flags[x]; flags[x] = flags[y]; flags[y] = temp; } /** * 排序 * @param flags * @return */ public String ChessSort(char[] flags){ int b = 0, w = 0; int r = flags.length - 1; while(w <= r){ switch(flags[w]){ case 'b': Swap(flags, b, w); b++; w++; break; case 'w': w++; break; case 'r': while(r < w && flags[r] == 'r'){ r--; } Swap(flags, r, w); r--; break; default: break; } } return new String(flags); } }