自己的思路(循环两遍)
循环一遍,记录0 1 2各自的个数,然后在循环一遍进行赋值
先回顾一下二切分问题的解法->单指针(数组访问一遍)
一个指针指向开头,依次遍历每个元素,如果是0则与指针位置的值交换并且指针后移
官方解法1->单指针(数组访问两遍)
使用二切分两次:先把0归位,再把1归位,最后2的位置就好了
class Solution {
public:
void sortColors(vector& nums) {
//单指针写法
int index = 0, n = nums.size();
for (int i = 0; i < n; ++i) {
if (nums[i] == 0) {
swap(nums[i], nums[index++]);
}
}
for (int i = index; i < n; ++i) {
if (nums[i] == 1) {
swap(nums[i], nums[index++]);
}
}
}
};
官方解法2->双指针(数组访问一遍)
一个指针指向开头,一个指针指向结尾,index从前往后遍历一遍,如果是0与头指针交换头指针后移index++,2与尾指针交换尾指针前移 index不能++
易错点:
当前元素是2时,与尾指针元素交换以后index不能++,因为有可能尾指针指向的是0!!!
class Solution {
public:
void sortColors(vector& nums) {
//双指针写法
int n = nums.size();
int front = 0, back = n - 1;
int index = 0;
while (index <= back) {
if (nums[index] == 0) {
swap(nums[index], nums[front++]);
index++;
} else if (nums[index] == 2){
swap(nums[index], nums[back--]);
}else {
index++;
}
//index++; //当nums[index]==2时,index不能++!!!
}
}
};



