题目描述
给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库内置的 sort 函数的情况下解决这个问题。
荷兰国旗问题与上述题目类似,可用同套代码解答
问题描述:
给定一个数组nums,和一个数target,请把小于等于target的数放在数组的左边,大于target的数放在数组的右边。要求额外空间复杂度O(1),时间复杂度 O(N)
思路
准备三个指针
第一个指针small_p 指向比target小的区域
第二个指针big_p 指向比target大的区域
第三个指针move_p 指向数组起始位置,用于遍历数组
数组左右各一个指针,分别代表小区域和大区域,通过move_p和target比较的情况,不停的让左右指针向中间移动
如果move_p < target small_p向后移动一位,然后交换small_p 和 move_p,再让move_p向后移动一位。
如果move_p = target 只需要把move_p 向后移动一位即可
如果move_p > target 把big_p向前移动一位,然后交换big_p 和 move_p
vector<int> france_flag(vector<int>& nums, int target)
{
//小于target指针
int small_p = -1;
//大于target指针
int big_p = nums.size();
//遍历数组的指针
int move_p = 0;
while(move_p < big_p)
{
if(nums[move_p] < target)
{
small_p++;
swap(nums, move_p, small_p);
move_p++;
}
else if(nums[move_p] == target)
{
move_p++;
}
else
{
big_p--;
swap(nums, move_p, big_p);
}
}
return nums;
}