Shruti Bibra Added to sort_colors as ShrutinewSort_Colors

C++/ShrutinewSort_Colors.cpp

@@ -0,0 +1,56 @@
+//https://leetcode.com/problems/sort-colors/
+
+#include <iostream>
+using namespace std;
+
+void sort012(int* arr, int n)
+{
+    // Variables to maintain the count of 0's,
+    // 1's and 2's in the array
+    int count0 = 0, count1 = 0, count2 = 0;
+    for (int i = 0; i < n; i++) {
+        if (arr[i] == 0)
+            count0++;
+        if (arr[i] == 1)
+            count1++;
+        if (arr[i] == 2)
+            count2++;
+    }
+
+    // Putting the 0's in the array in starting.
+    for (int i = 0; i < count0; i++)
+        arr[i] = 0;
+
+    // Putting the 1's in the array after the 0's.
+    for (int i = count0; i < (count0 + count1); i++)
+        arr[i] = 1;
+
+    // Putting the 2's in the array after the 1's
+    for (int i = (count0 + count1); i < n; i++)
+        arr[i] = 2;
+
+    return;
+}
+
+// Prints the array
+void printArray(int* arr, int n)
+{
+    for (int i = 0; i < n; i++)
+        cout << arr[i] << " ";
+    cout << endl;
+}
+
+// Driver code
+int main()
+{
+    int arr[] = { 0, 1, 1, 0, 1, 2, 1, 2, 0, 0, 0, 1 };
+    int n = sizeof(arr) / sizeof(arr[0]);
+    sort012(arr, n);
+    printArray(arr, n);
+    return 0;
+}
+
+/*
+* Hint : If asked to sort array of 3 numbers(elements) only
+* Variation of dutch national flag problem
+*/