[leetcode][C++] sort colors

https://leetcode.com/problems/sort-colors/

Sort Colors - LeetCode

 

 

[leetcode][C++] sort colors

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int low = 0;
        int high = nums.size() - 1;

        for (int i = 0; i <= high;) {
            if (nums[i] == 0) {
                swap(nums[i], nums[low]);
                low++; i++;
            }
            else if (nums[i] == 2) {
                swap(nums[i], nums[high]);
                high--;
            }
            else i++;
        }
    }
};