友情支持
如果您觉得这个笔记对您有所帮助，看在D瓜哥码这么多字的辛苦上，请友情支持一下，D瓜哥感激不尽，😜
|
|
有些打赏的朋友希望可以加个好友，欢迎关注D 瓜哥的微信公众号，这样就可以通过公众号的回复直接给我发信息。
公众号的微信号是: jikerizhi。因为众所周知的原因，有时图片加载不出来。 如果图片加载不出来可以直接通过搜索微信号来查找我的公众号。 |
75. 颜色分类
给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库内置的 sort 函数的情况下解决这个问题。
示例 1：
输入：nums = [2,0,2,1,1,0] 输出：[0,0,1,1,2,2]
示例 2：
输入：nums = [2,0,1] 输出：[0,1,2]
提示：
-
n == nums.length -
1 <= n <= 300 -
nums[i]为0、1或2
进阶：
-
你能想出一个仅使用常数空间的一趟扫描算法吗？
思路分析
经典的荷兰国旗问题，最合适的解法是：“三路快排”。
计数排序：统计 0、 1 和 2 的个数。然后在数组中依次填入。
-
一刷
-
二刷
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
/**
* Runtime: 0 ms, faster than 100.00% of Java online submissions for Sort Colors.
*
* Memory Usage: 35.2 MB, less than 99.21% of Java online submissions for Sort Colors.
*
* @author D瓜哥 · https://www.diguage.com
* @since 2019-10-29 20:26
*/
public void sortColors(int[] nums) {
if (Objects.isNull(nums) || nums.length <= 1) {
return;
}
int current = 0;
int head = 0;
int tail = nums.length - 1;
while (current <= tail) {
if (nums[current] == 0) {
swap(nums, current, head);
head++;
current++;
} else if (nums[current] == 2) {
swap(nums, current, tail);
tail--;
// 这了不需要 current++，就是把这个数字重新排队一遍！
} else {
current++;
}
}
}
private void swap(int[] nums, int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
/**
* @author D瓜哥 · https://www.diguage.com
* @since 2025-11-17 21:37:31
*/
public void sortColors(int[] nums) {
int[] counter = new int[3];
for (int num : nums) {
counter[num]++;
}
int index = 0;
for (int i = 0; i < counter.length; i++) {
for (int j = 0; j < counter[i]; j++) {
nums[index++] = i;
}
}
}