Steward

分享是一種喜悅、更是一種幸福

程式語言 - LeetCode - C++ - 75. Sort Colors

---

題目：


解答：

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

        while (i <= r) {
            if (nums[i] == 0) {
                swap(nums[i++], nums[l++]);
            }
            else if (nums[i] == 2) {
                swap(nums[i], nums[r--]);
            }
            else {
                i += 1;
            }
        }
    }
};