给定一个包含红色、白色和蓝色、共
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
进阶:
- 你能想出一个仅使用常数空间的一趟扫描算法吗?
思路:
本题是经典的「荷兰国旗 问题」:现在有若干个红、白、蓝三种颜色的球随机排列成一条直线。现在我们的任务是把这些球按照红、白、蓝排序。
参考 三指针法:LeetCode题解
0,1,2 排序。一次遍历,若是0,则移动到表头,若是2,则移动到表尾。不用考虑1。0和2处理完,1还会有错吗?
- for循环判断条件:本来是 i <= len(nums),但right指针右侧已经都是2了,没必要继续寻找。因此以越过右指针为终止条件,减少查找次数
- 判断p所在索引的数字
- 找到0就与 left 头指针交换
- 找到2就与 right 尾指针交换
- 找到1则跳过
时间复杂度:O(N),遍历数组
空间复杂度:O(1),原地修改
// 参考 三指针法:https://leetcode.cn/problems/sort-colors/solutions/1947033/by-zyh698-e7u1/?languageTags=golang
// 荷兰国旗问题:0,1,2 排序。一次遍历,若是0,则移动到表头,若是2,则移动到表尾。不用考虑1。0和2处理完,1还会有错吗?
// left左侧都是0,right右侧都是2,左右侧都确定好了,那么中间的就自然全是1了
func sortColors(nums []int) {
left, right, i := 0, len(nums)-1, 0
// 本来是 i <= len(nums),但right指针右侧已经都是2了,没必要继续寻找。因此以越过右指针为终止条件,减少查找次数
for i <= right {
// 找到0就与左指针l交换
if nums[i] == 0 {
nums[left], nums[i] = nums[i], nums[left]
left++
i++
// nums[l]已经在i向右遍历的过程中早就验证过了,所以i要加加(注意:left<=i,left在i的右侧)
// 找到1就跳过
} else if nums[i] == 1 {
i++
// 找到2就与右指针r交换
} else { // if nums[i] == 2
nums[i], nums[right] = nums[right], nums[i]
right--
// 此时i不用加1,因为nums[right]交换过来的数还没有验证,有可能是0或2
}
}
}