这道荷兰旗问题，我面试时遇到三次！

最新推荐文章于 2024-11-04 19:24:37 发布

吴师兄学算法

最新推荐文章于 2024-11-04 19:24:37 发布

阅读量286

点赞数 1

算法字节跳动 powerdesigner ext sms

大家好，我是不会写代码的小浩。今天为大家分享经典的荷兰国旗问题。

01、题目示例

"荷兰国旗问题" 是计算机科学中的一个经典题目，它是由Edsger Dijkstra提出的。荷兰国旗由红、白、蓝三色组成。

荷兰国旗问题：现在有若干个红、白、蓝三种颜色的球随机排列成一条直线。现在我们的任务是把这些球按照红、白、蓝排序。
这个问题之所以叫荷兰国旗，是因为我们可以将红白蓝三色小球想象成条状物，有序排列后正好组成荷兰国旗。

大概就是这么个意思：

02、题解分析

这道题很经典，很高频。

便于分析，我们把上面的图稍微改一下：

改成这样：

我们很容易可以想到的是，最终排序完成后的数组是分成三份的：

红-白-蓝

那总共就三个颜色，我们要区分开来，是不是最少需要两条分隔线？A线的左侧为0，右侧为1。B线的左侧为1，右侧为2。

但是刚开始的时候，红-白-蓝三色是乱序的，所以此时的两条线我们是不是可以看成在最两侧？

那我们剩下的是不是只需要把 A线和 B线间的数据维护成满足 AB 线的规则就可以了？那要维护 AB 线间的数据，是不是至少你得遍历下 AB 线间的数据？

我们从 C 位置处开始，我们发现此时 C 等于0。是不是意味着，我们应把这个元素放到 A 的左侧，所以我们移动 A线。当然，我们也需要移动一下 C 的位置。（CASE-1）

然后我们发现新的 C 位置处等于2，那是不是说明这个元素应该位于 B 的右侧。所以我们要把该位置的元素和 B位置处的元素进行交换，同时移动B。(CASE-2)

但是这里要注意，C 交换完毕后，C 不能向前移。因为C指向的元素可能是属于前部的，若此时 C 前进则会导致该位置不能被交换到前部。继续向下遍历。

有意思了，我们发现 C 指向位置处等于1。有没有发现这种本身就满足规则了？所以我们忽略就可以了。（CASE-3） 继续移动 C。

主要就这三种 CASE，我们把剩下的图都绘制出来：

总结一下：

1）若遍历到的位置为0，则说明它一定位于A的左侧。于是就和A处的元素交换，同时向右移动A和C。
2）若遍历到的位置为1，则说明它一定位于AB之间，满足规则，不需要动弹。只需向右移动C。
3）若遍历到的位置为2，则说明它一定位于B的右侧。于是就和B处的元素交换，交换后只把B向左移动，C仍然指向原位置。（因为交换后的C可能是属于A之前的，所以C仍然指向原位置）

大概就是这么一个分析过程，代码其实就很简单了：

python 版本：

//py3 
class Solution: 
    def sortColors(self, nums: List[int]) -> None: 
        a = c = 0 
        b = len(nums) - 1 
        while c <= b:
            if nums[c] == 0: 
                nums[a], nums[c] = nums[c], nums[a]
                a += 1
                c += 1
            elif nums[c] == 2:
                nums[c], nums[b] = nums[b], nums[c]
                b -= 1
            else:
                c += 1

go 版本：

//go 
func sortColors(nums []int) { 
    a := 0 
    b := len(nums) - 1 
    for c := 0; c <= b; c++ { 
        if nums[c] == 0 { 
            nums[c], nums[a] = nums[a], nums[c] 
            a++
        }
        if nums[c] == 2 {
            nums[c], nums[b] = nums[b], nums[c]
            c--
            b--
        }
    }
}

执行结果：