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]
进阶:
- 你可以不使用代码库中的排序函数来解决这道题吗?
- 你能想出一个仅使用常数空间的一趟扫描算法吗?
思路&代码
这道题的要注意的就是不使用sort函数并且要进行原地排序。
使用双指针 p 0 p_0 p0 和 p 1 p_1 p1 分别记录当前应交换 0 和 1 的位置,即 p 0 p_0 p0 之前都是 0, p 0 p_0 p0 到 p 1 p_1 p1 之间都是 1, p 1 p_1 p1 及之后都是 2。注意,必有 p 0 ≤ p 1 p_0 \leq p_1 p0≤p1。
遍历列表 nums
,当遍历到第 i 个位置时:
- 当 n u m s [ i ] = 2 nums[i] = 2 nums[i]=2 时,什么也不做。
- 当 n u m s [ i ] = 1 nums[i] = 1 nums[i]=1 时,将 n u m s [ i ] nums[i] nums[i] 与 n u m s [ p 1 ] nums[p_1] nums[p1] 互换,并将 p 1 p_1 p1 后移一位。这时交换必然是值为 1 和 2 的交换,没有什么特殊情况。
- 当
n
u
m
s
[
i
]
=
0
nums[i] = 0
nums[i]=0 时,将
n
u
m
s
[
i
]
nums[i]
nums[i] 与
n
u
m
s
[
p
0
]
nums[p_0]
nums[p0] 互换。需要注意的是,
(1) 如果 p 0 = p 1 p_0 = p_1 p0=p1,则说明 还没遇到第二种情况(即 p 1 p_1 p1还未+1),则此时 i 后面的 1 还没换到前面来,也就是说 n u m s [ p 0 ] = n u m s [ p 1 ] = 2 nums[p_0] = nums[p_1] = 2 nums[p0]=nums[p1]=2,这时直接换完后把 p 0 p_0 p0 和 p 1 p_1 p1 后移一位即可,因为是值为 0 和 2 互换,这很正常;
(2) 如果 p 0 < p 1 p_0 < p_1 p0<p1,则说明已经遇到过第二种情况( p 1 p_1 p1已经后移,与 p 0 p_0 p0分开),此时 n u m s [ p 0 ] = 1 nums[p_0] = 1 nums[p0]=1,这时将 n u m s [ i ] nums[i] nums[i] 与 n u m s [ p 0 ] nums[p_0] nums[p0] 互换,则是值为 1 和 0 的互换,这时原本之前换到前面的 1 又换到后面去了,即换后 n u m s [ i ] = 1 nums[i] = 1 nums[i]=1(先前是 n u m s [ i ] = 0 nums[i] = 0 nums[i]=0),就需要再把这个 1 和前面的 2 换一下,也就是 n u m s [ i ] nums[i] nums[i] 与 n u m s [ p 1 ] nums[p_1] nums[p1] 互换,然后把 p 0 p_0 p0 和 p 1 p_1 p1 后移一位。
举个例子:
输入:nums = [2,0,2,1,1,0]
输出:[0,0,1,1,2,2]
通过上面这个步骤每一步得到的结果为:
![](https://img-blog.csdnimg.cn/923089d80e6244268518d19fa1d0d728.png)
![](https://img-blog.csdnimg.cn/8ed3b5559a014e1a85bae91a7890d606.png)
![](https://img-blog.csdnimg.cn/f9a35a2304af4ccf99c366988f8779da.png)
![](https://img-blog.csdnimg.cn/79113c118ddf46d19423b9e6ea9defb5.png)
![](https://img-blog.csdnimg.cn/f44f4431fa7348f98700b40aafdba629.png)
这步以后其实就应该注意了, p 0 p_0 p0 和 p 1 p_1 p1 分开了,再遇到 0 的时候就不仅仅是交换 n u m s [ i ] nums[i] nums[i] 与 n u m s [ p 0 ] nums[p_0] nums[p0] 这么简单了。
![](https://img-blog.csdnimg.cn/84745413b0ee40e99b7a5eca070d072b.png)
![](https://img-blog.csdnimg.cn/610c64e31eb1404883f759dc6c942b6f.png)
![](https://img-blog.csdnimg.cn/d845e176fd184d278d8acfb6225ef888.png)
最后这步可以分成两个子步来解释,就是上面描述的第三种情况的(2), n u m s [ i ] nums[i] nums[i] 与 n u m s [ p 0 ] nums[p_0] nums[p0]交换后我们却把先前换到前面的1有放到后面去了,这里就要再把那个1放回到前面,于是就要再交换下 n u m s [ i ] nums[i] nums[i] 和 n u m s [ p 1 ] nums[p_1] nums[p1]:
![](https://img-blog.csdnimg.cn/dfb44ce1043e4fc388b88fd4bb0005d9.png)
class Solution:
def sortColors(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
p0 = 0
p1 = 0
for i in range(len(nums)):
if nums[i] == 0:
nums[p0], nums[i] = nums[i], nums[p0]
if p0 < p1:
# 说明上步交换的是0和1,实际想要的是0和2交换,所以这里再把1(之前是0)和2交换
nums[p1], nums[i] = nums[i], nums[p1]
p0 += 1
p1 += 1
elif nums[i] == 1:
nums[p1], nums[i] = nums[i], nums[p1]
p1 += 1
return nums