三色旗问题通解

背景描述：

描述：假设有一条绳子，上面挂有红、白、蓝三种颜色的旗子，没有顺序。

要求：将旗子分类，按蓝、白、红的顺序排列。

问题：如何移动，可使总移动次数最少。

限制：只能在绳子上操作，每次只能交换其中两面旗子。

例子：

原始状态：

最终状态：

该问题是由荷兰人Dijkstra提出的，所以该问题又叫荷兰旗问题。

 

解析：

一般讲到这种问题，大家都不自然地会想到部分快排。先来看种简单的快排写法：

func QuickSort(slice []int) {
	if len(slice) <= 1 {
		return
	}

	t := slice[0]
	l := 0
	r := len(slice) - 1
	for l < r {
		for ; l < r && slice[r] > t; r-- {
		}
		for ; l < r && slice[l] <= t; l++ {
		}
		slice[l], slice[r] = slice[r], slice[l] // swap
	}
	slice[l], slice[0] = slice[0], slice[l] // swap
	QuickSort(slice[:l])
	QuickSort(slice[l+1:])
}

简单粗暴的写法。

网上盛行的三色题解法就是根据上面写法得来的。

 

我们先把问题简化，将三色旗改成双色旗：

1

2

3

4

5

6

按上面快排的思路，就是在右端找一个蓝色的（5），在左端找一个红色的（1），调换顺序：

5

2

3

4

1

6

重复这个过程，从右端找到一个蓝色的（4），左端找到一个红色的（3），调换顺序：