diff --git a/Sort Colour/cppSolutionSortColour.cpp b/Sort Colour/cppSolutionSortColour.cpp new file mode 100644 index 0000000..c9f0b91 --- /dev/null +++ b/Sort Colour/cppSolutionSortColour.cpp @@ -0,0 +1,20 @@ +//https://leetcode.com/problems/sort-colors/ + +class Solution { +public: + void sortColors(vector& arr) { + int mid=0; + int i=0,j=arr.size()-1; + while(mid<=j){ + if(arr[mid]==0){ + swap(arr[i],arr[mid]); + i++;mid++; + }else if(arr[mid]==1){ + mid++; + }else if(arr[mid]==2){ + swap(arr[j],arr[mid]); + j--; + } + } + } +};