diff --git a/12 June Count numbers containing 4 b/12 June Count numbers containing 4 new file mode 100644 index 0000000..b480746 --- /dev/null +++ b/12 June Count numbers containing 4 @@ -0,0 +1,14 @@ +class Solution { +public: + void sortColors(vector& nums) { + int r = 0, w = 0, b = nums.size() - 1; + while (w <= b){ + if (!nums[w]) + swap(nums[r++], nums[w++]); + else if (nums[w] == 1) + w++; + else + swap(nums[w], nums[b--]); + } + } +};