# Sort an array of 0s, 1s and 2s (Simple Counting)

Given an array A[] consisting 0s, 1s and 2s, write a function that sorts A[]. The functions should put all 0s first, then all 1s and all 2s in last.
Examples:

```Input :  {0, 1, 2, 0, 1, 2}
Output : {0, 0, 1, 1, 2, 2}

Input :  {0, 1, 1, 0, 1, 2, 1, 2, 0, 0, 0, 1}
Output : {0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 2, 2}

```

Count the number of 0’s, 1’s and 2’s. After Counting, put all 0’s first, then 1’s and lastly 2’s in the array. We traverse the array two times. Time complexity will be O(n).

## CPP

 `// Simple C++ program to sort an array of 0s` `// 1s and 2s.` `#include ` `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);` `    ``sort012(arr, n);` `    ``printArray(arr, n);` `    ``return` `0;` `}`

## Java

 `// Simple Java program` `// to sort an array of 0s` `// 1s and 2s.` `import` `java.util.*;` `import` `java.lang.*;`   `public` `class` `GfG{`   `    ``public` `static` `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``;` `        ``int` `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``;` `    `  `        ``printArray(arr, n);` `    ``}` `    `  `    ``// Prints the array` `    ``public` `static` `void` `printArray(``int` `arr[], ``int` `n)` `    ``{` `        ``for` `(``int` `i = ``0``; i < n; i++)` `            ``System.out.print(arr[i] + ``" "``);` `        ``System.out.println();` `    ``}` `    `  `    ``// Driver function ` `    ``public` `static` `void` `main(String argc[])` `    ``{` `    `  `        ``int` `arr[] = { ``0``, ``1``, ``1``, ``0``, ``1``, ``2``, ``1``,` `                    ``2``, ``0``, ``0``, ``0``, ``1` `};` `        ``int` `n = ``12``;` `        ``sort012(arr, n);` `    ``}` `}`   `// This code is contributed by Sagar Shukla `

## Python3

 `# Python C++ program to sort an array of 0s` `# 1s and 2s.` `import` `math`   `def` `sort012(arr, n):`   `    ``# Variables to maintain the count of 0's, ` `    ``# 1's and 2's in the array` `    ``count0 ``=` `0` `    ``count1 ``=` `0` `    ``count2 ``=` `0` `    ``for` `i ``in` `range``(``0``,n):` `        ``if` `(arr[i] ``=``=` `0``):` `            ``count0``=``count0``+``1` `        ``if` `(arr[i] ``=``=` `1``):` `            ``count1``=``count1``+``1` `        ``if` `(arr[i] ``=``=` `2``):` `            ``count2``=``count2``+``1` `    `    `    ``# Putting the 0's in the array in starting.` `    ``for` `i ``in` `range``(``0``,count0):` `        ``arr[i] ``=` `0` `    `  `    ``# Putting the 1's in the array after the 0's.` `    ``for` `i ``in` `range``( count0, (count0 ``+` `count1)) :` `        ``arr[i] ``=` `1` `    `  `    ``# Putting the 2's in the array after the 1's` `    ``for` `i ``in` `range``((count0 ``+` `count1),n) :` `        ``arr[i] ``=` `2` `    `  `    ``return`     `# Prints the array` `def` `printArray( arr,  n):`   `    ``for` `i ``in` `range``(``0``,n):` `        ``print``( arr[i] , end``=``" "``)` `    ``print``()`     `# Driver code` `arr ``=` `[ ``0``, ``1``, ``1``, ``0``, ``1``, ``2``, ``1``, ``2``, ``0``, ``0``, ``0``, ``1` `]` `n ``=` `len``(arr)` `sort012(arr, n)` `printArray(arr, n)`   `# This code is contributed by Gitanjali.`

## C#

 `// Simple C# program` `// to sort an array of 0s` `// 1s and 2s.` `using` `System;`   `public` `class` `GfG{`   `    ``public` `static` `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;` `        ``int` `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;` `    `  `        ``printArray(arr, n);` `    ``}` `    `  `    ``// Prints the array` `    ``public` `static` `void` `printArray(``int` `[]arr, ``int` `n)` `    ``{` `        ``for` `(``int` `i = 0; i < n; i++)` `            ``Console.Write(arr[i] + ``" "``);` `        ``Console.WriteLine();` `    ``}` `    `  `    ``// Driver function ` `    ``public` `static` `void` `Main()` `    ``{` `    `  `        ``int` `[]arr = { 0, 1, 1, 0, 1, 2, 1,` `                    ``2, 0, 0, 0, 1 };` `        ``int` `n = 12;` `        ``sort012(arr, n);` `    ``}` `}`   `// This code is contributed by vt_m`

Output

```0 0 0 0 0 1 1 1 1 1 2 2

```

Output:

```0 0 0 0 0 1 1 1 1 1 2 2

```

Problems with the above solution.:

1.  It requires two traversals of array.
2. This solution may not work if values are a part of the structure. For example, consider a situation where 0 represents Computer Science Stream, 1 represents Electronics and 2 represents Mechanical. We have a list of student objects (or structures) and we want to sort them. We cannot use the above sort as we simply put 0s, 1s, and 2s one by one.

Another Approach:

## Python3

 `# Example` `#` `# input = [0, 1, 2, 2, 0, 0]` `# output = [0, 0, 0, 1, 2, 2]`   `inputArray ``=` `[``0``, ``1``, ``1``, ``0``, ``1``, ``2``, ``1``, ``2``, ``0``, ``0``, ``0``, ``1``]` `outputArray ``=` `[]` `indexOfOne ``=` `0` `for` `item ``in` `inputArray:` `    ``if` `item ``=``=` `2``:` `        ``outputArray.append(item)` `    ``elif` `item ``=``=` `1``:` `        ``outputArray.insert(indexOfOne, item)` `        ``indexOfOne ``+``=` `1` `    ``elif` `item ``=``=` `0``:` `        ``outputArray.insert(``0``, item)` `        ``indexOfOne ``+``=` `1` `    ``else``:` `        ``print``(``" wrong value - Aborting "``)` `        ``continue` `print``(outputArray)`

Output

```[0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2]

```

Optimal Solution that handles above issues :
Sort an array of 0s, 1s and 2s (Dutch National Flag Algorithm)

Attention reader! Don’t stop learning now. Get hold of all the important DSA concepts with the DSA Self Paced Course at a student-friendly price and become industry ready.

My Personal Notes arrow_drop_up Check out this Author's contributed articles.

If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.

Improved By : kishandewangan

Article Tags :
Practice Tags :

11

Please write to us at contribute@geeksforgeeks.org to report any issue with the above content.