题目描述:
难度:中等
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶:
- 一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。 - 你能想出一个仅使用常数空间的一趟扫描算法吗?
思路:
1、首先这题是让你一趟排序,区分三种不同的数字,所以如果仅仅只用两个指针,那肯定是解决不了的,所以使用三个指针。
2、三个指针的含义:既然我们要区分三种不同的数字,所以我们用两个指针来管理两边的数字界限:
- left 是0的右界,即只要下标小于left 的元素都是0。
- right是2的左界,即大于下标rigth 的元素都是2。
- mid 用来遍历之中的元素和两边进行交换。
- 如果是1则跳过。
- 如果是0则 和 left 元素交换,并且 left++,mid++。 此时nums[left]的值一定是1, 因为如果是2则进行交换了,如果是0,left++。
- 如果是2则 和 right 元素交换,right--。
// 使用三个指针来解决 荷兰国旗问题 第一指针之前的元素全部为0 第三个指针的元素之后全部为2 中间指针用来遍历进行交换。
class Solution {
public:
void sortColors(vector<int>& nums) {
int left = 0, right = nums.size()-1, mid=0;
// while(left<=right && nums[left] == 0) left++;
// while(left<= right && nums[right] == 2) right--;
// mid = left;
while(mid <= right){
if(nums[mid] == 1) mid++;
else if(nums[mid] == 0) swap(nums[left++], nums[mid++]);
else swap(nums[mid], nums[right--]);
}
}
};