双指针法解决荷兰国旗问题：颜色分类算法详解

1. 题目背景与核心需求

这道力扣HOT100系列的第75题"颜色分类"是一个经典的荷兰国旗问题变种。题目要求我们将一个仅包含0、1、2的数组原地排序，使得所有0排在前面，1在中间，2在最后。看似简单的需求背后隐藏着对双指针技巧和快速排序思想的深度考察。

在实际工程中，类似的三分类问题非常常见。比如电商系统中的订单状态筛选（待付款/已发货/已完成）、日志系统的错误等级分类（info/warning/error）等场景。这道题的特殊性在于要求：

• 只能使用常数空间（不能新建数组）
• 必须一趟扫描完成（时间复杂度O(n)）
• 需要处理元素交换的边界条件

2. 算法思路解析

2.1 双指针法的设计哲学

传统排序算法如快速排序虽然能解决问题，但无法满足O(n)时间复杂度的要求。荷兰国旗问题的经典解法是使用三指针：

• left指针：标记0的右边界（不包含）
• right指针：标记2的左边界（不包含）
• current指针：当前遍历位置

这种设计实现了：

1. 空间复杂度O(1)：仅使用几个指针变量
2. 时间复杂度O(n)：单次遍历即可完成
3. 原地操作：直接在原数组交换元素

2.2 指针移动规则详解

具体操作时需要特别注意指针的移动逻辑：

python复制def sortColors(nums):
    left, current, right = 0, 0, len(nums) - 1
    
    while current <= right:
        if nums[current] == 0:
            nums[left], nums[current] = nums[current], nums[left]
            left += 1
            current += 1
        elif nums[current] == 2:
            nums[current], nums[right] = nums[right], nums[current]
            right -= 1
        else:
            current += 1

关键点解析：

1. 遇到0时：与left交换后两个指针都右移（因为从左交换来的只可能是1）
2. 遇到2时：只移动right指针（因为从右交换来的可能是0/1/2）
3. 遇到1时：直接跳过

3. 边界条件与易错点

3.1 指针越界防护

实际编码中最容易出现的错误是：

• right指针初始值应为len(nums)-1而非len(nums)
• while循环条件必须是current <= right（包含等号）
• 交换2时current不能自增（需要检查交换来的新元素）

3.2 测试用例设计建议

完整的测试应该包含：

python复制test_cases = [
    ([2,0,1], [0,1,2]),        # 基础乱序
    ([2,2,2], [2,2,2]),        # 全2场景
    ([0,0,0], [0,0,0]),        # 全0场景
    ([1,1,1], [1,1,1]),        # 全1场景
    ([2,0,2,1,1,0], [0,0,1,1,2,2])  # 混合复杂案例
]

4. 算法优化与变种

4.1 计数排序法的取舍

虽然计数排序可以实现：

python复制def sortColors(nums):
    count = [0]*3
    for num in nums:
        count[num] += 1
    nums[:] = [0]*count[0] + [1]*count[1] + [2]*count[2]

但这种方法：

• 需要两次遍历（统计+重建）
• 不符合"原地修改"的严格题意
• 在元素种类增多时扩展性差

4.2 多分类场景扩展

当需要处理更多类别时（如0-4共5类），可以采用：

1. 双指针向中间扩展为多指针
2. 使用快速排序的partition思想递归处理
3. 彩虹排序(Rainbow Sort)等高级算法

5. 工程实践中的注意事项

在实际项目中使用类似算法时需要注意：

1. 如果元素是对象而非简单数字，交换成本可能很高
2. 多线程环境下需要加锁保护整个排序过程
3. 可以考虑使用STL的partition函数（C++）或numpy的partition（Python）

关键经验：当遇到需要保持相对顺序的类似问题时，应该考虑使用stable partition算法而非简单的双指针交换。

6. 复杂度分析与对比

7. 刷题技巧与学习路径

这道题在力扣题库中的定位是中等难度，但实际考察点非常典型。建议学习者：

1. 先手写5遍完整代码，体会指针移动规律
2. 用纸笔模拟[2,0,2,1,1,0]的排序过程
3. 尝试修改为四分类问题（增加数字3）
4. 对比练习类似题目：
   • 移动零（283题）
   • 按奇偶排序数组（905题）
   • 自定义字符串排序（791题）

我在最初练习时最容易犯的错误是在交换2之后又立即移动current指针，导致某些0被跳过。后来通过打印每次交换后的数组状态，才彻底理解了指针移动的边界条件。建议大家在遇到类似问题时，一定要用print调试法观察中间状态。