leetcode-----75. 颜色分类

• 链接：https://leetcode-cn.com/problems/sort-colors/

代码

class Solution {
public:
    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++]);
            else if (nums[i] == 2) swap(nums[i], nums[k--]);
            else i++; 
        }   
    }
};