设L,R为左右第一个不符合要求的点。cur为当前遍历的点,遍历顺序为从左往右。
当num[cur]==1时,cur++。
当num[cur]==0时,交换cur和L对应的数,然后cur++,L++。
当num[cur]==2时,交换cur和R对应的数,R–。
如果cur>L,说明L指向的数肯定是1,所以当num[cur] == 0时,交换cur和L后,两者都要++。
如果cur==L,如果num[cur]==0,此时两者都++是显然的。
如果num[cur]==2,此时num[R]指向的数是不确定的,所以此时交换cur和R后cur不用++。
class Solution {
public:
void sortColors(vector<int>& nums) {
int l=0,r=nums.size()-1;
int cur=0;
while(cur<=r){
if(nums[cur]==0)swap(nums[l++], nums[cur++]);
else if(nums[cur]==2) swap(nums[r--], nums[cur]);
else ++cur;
}
}
};