Challenge: Partitioning of 0s, 1s, and 2s

Learn how you can solve the problem of partitioning 0s, 1s, and 2s using a sorting algorithm.

Problem

Given an array containing 0s, 1s, and 2s, write a program to sort the array so that 0s come first, followed by 1s, and then 2s in the end.

Input

An integer array.

Output

An integer array.

Sample input

array = { 0, 0, 2, 0, 2, 1, 0, 1 }

Sample output

array = { 0, 0, 0, 0, 1, 1, 2, 2 }

Coding exercise

Try to solve this yourself first. If you get stuck or need help, you can always press the ā€œShow Solutionā€ button to see how your problem can be solved. Weā€™ll look at the solution in the next lesson.

Good luck!

Level up your interview prep. Join Educative to access 80+ hands-on prep courses.