基于快排的思想
回忆一下快排的思路:
把第一个位置的作为pivot,从右遍历直到发现有比pivot小的,交换;从左遍历,直到发现有比pivot大的,交换
注意这里交换的都是当前的low和high,直到low,high相遇,本轮停止
如果数字里面只有0,1,2那么显然只需要一次while就可以结束
思路依旧是l r current
current==0 需要和low交换,使得0都在左侧
current==1 没问题,毕竟现在左边的都是0,右边的都是2。继续遍历
current==2 需要和high交换,使得2都在右侧
归根结底还是对于快排的思想掌握的不好
class Solution {
public:
void sortColors(vector<int>& nums) {
int l=0,r=nums.size()-1;
int current=0;
while(current<=r){
if(nums[current]==0){
swap(nums[current],nums[l]);
l++;
current++;
}
else if(nums[current]==1)
current++;
else{
swap(nums[current],nums[r]);
r--;
}
}
}
};