技巧

颜色分类

中等#75Java时间 O(n)空间 O(1)

题目要做什么

数组只包含 0、1、2,分别表示三种颜色。一次扫描、原地排序,让相同颜色相邻并按 0、1、2 排列。

题目示例输入nums = [2,0,2,1,1,0]
题目示例输出[0,0,1,1,2,2]

01 为什么这样做

荷兰国旗法维护三个区域:左侧已放好的 0,中间已确认的 1,右侧已放好的 2。遇到 2 时与右侧交换后,当前位置仍未知,不能立即前进。

始终成立的条件

[0, low) 是 0,[low, i) 是 1,(high, n) 是 2,[i, high] 未处理。

02 看见算法执行

动画与代码同步

修改输入,播放自己的例子

当前使用题目示例
修改输入会改变执行过程;预期输出用于核对结果。
数组 / nums步骤 1
执行状态
本次演示输出播放到最后查看

Solution.java参考解法
1class Solution {2    public void sortColors(int[] nums) {3        int low = 0, i = 0, high = nums.length - 1;4        while (i <= high) {5            if (nums[i] == 0) {6                swap(nums, i, low);7                low++; i++;8            } else if (nums[i] == 2) {9                swap(nums, i, high);10                high--;11            } else i++;12        }13    }14    private void swap(int[] nums, int a, int b) {15        int temp = nums[a];16        nums[a] = nums[b];17        nums[b] = temp;18    }19}

键盘控制:A 后退 · D 前进

编程练习

↑↓ 选择↵ 打开Pagefind 全文检索