原题链接
维护如图所示性质,对 i 从左到右扫描:
①如果 nums[i] == 0:
则交换 j 与 i 的位置所在的数字,此时
为了维护性质,则需 j + 1、i + 1
②如果 nums[i] == 2:
则交换 i 与 k 的位置所在的数字,由于 k 的位置的数字不知道是 1 还是 0 ,所以只需要对 k - 1 即可,i 位置的数等待下一次循环。
③如果 nums[i] == 1:
进行 i + 1。
c++:
class Solution {
public:
int q[310];
void sortColors(vector<int>& nums) {
for(int i = 0,j = 0,k = nums.size() - 1;i <= k;)
{
if(nums[i] == 0) swap(nums[i],nums[j]),i++,j++;
else if(nums[i] == 2) swap(nums[i],nums[k]),k--;
else i++;
}
}
};