力扣:颜色分类

原题链接在这里插入图片描述
维护如图所示性质,对 i 从左到右扫描:
①如果 nums[i] == 0:
则交换 ji 的位置所在的数字,此时
![(https://img-blog.csdnimg.cn/3ae60b9272b7481b9834c083d52e4edf.png)
为了维护性质,则需 j + 1、i + 1
②如果 nums[i] == 2:
则交换 ik 的位置所在的数字,由于 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++;
       }
    }
};

猜你喜欢

转载自blog.csdn.net/K_CRACKING/article/details/130321599