题目
给定一个包含红色、白色和蓝色、共 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]
分析
三指针法
初始化指针:
left
指针初始化为 0,用于标记 0 应该存放的位置。right
指针初始化为nums.size() - 1
,用于标记 2 应该存放的位置。current
指针初始化为 0,用于遍历数组。
遍历数组:
- 当
current
指针小于等于right
指针时,进行以下操作: - 如果
nums[current]
等于 0,说明当前元素是红色,将其与left
位置的元素交换,并将left
和current
指针都向后移动一位。 - 如果
nums[current]
等于 2,说明当前元素是蓝色,将其与right
位置的元素交换,并将right
指针向前移动一位。注意,此时current
指针不移动,因为交换过来的元素还需要再次判断。 - 如果
nums[current]
等于 1,说明当前元素是白色,直接将current
指针向后移动一位。
结束条件:
- 当
current
指针大于right
指针时,遍历结束,数组已经按照 0、1、2 的顺序排列好。
时间复杂度:O(),
是数组的长度
空间复杂度:O(1)
class Solution {
public:void sortColors(std::vector<int>& nums) {int left = 0; // 指向 0 应该存放的位置int right = nums.size() - 1; // 指向 2 应该存放的位置int current = 0; // 当前遍历到的元素位置while (current <= right) {if (nums[current] == 0) {// 如果当前元素是 0,将其与 left 位置的元素交换std::swap(nums[current], nums[left]);left++;current++;} else if (nums[current] == 2) {// 如果当前元素是 2,将其与 right 位置的元素交换std::swap(nums[current], nums[right]);right--;} else {// 如果当前元素是 1,直接移动到下一个元素current++;}}}
};