三路快排的思想,省去了切分操作,直接一遍遍历。如果当前位是0,就替换到最左边,并把i和left都++,如果为2,就替换到最右边,并把right–,但是i不能++,因为替换过来的值没有检查过。如果为1就不变i++。直到I>right退出循环,等于不能退出,比如2,0,1,如果right等于i了,那么102直接退出了。
class Solution {
public:
void sortColors(vector<int>& nums) {
//用三路快排,1是切分点,0左侧,2右侧,1中间,快排核心思想=切分
//因为是原地排序所以肯定是快排
//扫描一遍,如果是0就放到前面,如果是2就放到后面,0212也适用
int i = 0,left = 0,right = nums.size()-1;
while(i <= right){
if(nums[i] == 0) swap(nums[i++],nums[left++]);
else if(nums[i] == 2) swap(nums[i],nums[right--]);//因为换过来的right还没检查过,所以要i还不能++
else i++;
}
}
};
class Solution {
public:
void sortColors(vector<int>& nums) {
//三路切分快排,直接swap即可,无需切分了
//双指针,将<1的放到左边,左边++,将>1的放到右边,右边--,但是自身不能++,因为右边换过来的还没检查过
int left = 0, right = nums.size()-1;
int i = 0;
//不用把数组全部遍历一遍,不然会把最后面的2又重新换到前面来
//为什么退出条件是i>right要搞清楚,一定要搞清楚
//当i>right的时候就会把2往前面换了,但是等于的时候还不能退出
//比如201这个例子,102时,i和right都指向0,但是i现在要和left换,所以当i和right相等的时候,这个数字还是要检查一下的,因为它有可能是要放到前面去的。
while(i <= right){
if(nums[i] < 1){
swap(nums[i++],nums[left++]);
}
else if(nums[i] > 1){
swap(nums[i],nums[right--]);
}
else{
i++;
}
}
}
};
三取样切分。注意循环结束的条件即可。
class Solution {
public:
void sortColors(vector<int>& nums) {
//原地排序,就是三取样切分
//碰到0就放到最前面,碰到1就继续往后走,碰到2就放到最后面
//如果从前面换过来的东西不用检查,后面换过来的要检查完再往后走
//何时结束循环要搞清楚
int l = 0, r = nums.size()-1, i = 0;
//i>r时退出,i=r不能退出,因为当前r这个数字没有被检查过
while(i <= r){
if(nums[i] == 1) i++;
else if(nums[i] == 0){
swap(nums[i],nums[l]);
i++;
l++;
}
else{
swap(nums[i],nums[r]);
r--;
}
}
}
};