75. Sort Colors
难度: Medium
刷题内容
原题连接
- https://leetcode-cn.com/problems/sort-colors/submissions/
内容描述
``` 给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意: 不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0] 输出: [0,0,1,1,2,2] 进阶：
一个直观的解决方案是使用计数排序的两趟扫描算法。 首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。 你能想出一个仅使用常数空间的一趟扫描算法吗？ ```
解题方案
思路 1
记录两个指针，一个是0部分的后一位，另一个是2开头的前一个位置。
遍历数组，遇到2，就和后一个指针交换（交换回来的有可能还是2，所以这个位置要再次判断）
遇到0则和最后一个数的下一个位置交换。
void sortColors(vector<int>& nums) {
int left = 0;
int right = nums.size()-1;
for(int i=0;i<=right;i++){
if(nums[i]==0){
int tmp = nums[left];
nums[left] = nums[i];
nums[i] = tmp;
left++;
}
else if(nums[i]==2){
int tmp = nums[i];
nums[i] = nums[right];
nums[right] = tmp;
right--;
i--;
}
}
}
