How should I practice Sort Colors?

Start by reading the full statement, solve the visible examples, then compare the pattern against related Capital One onsite interview problems in FastPrep.

Is Sort Colors associated with Capital One?

FastPrep tags this as a Capital One onsite interview coding practice problem when the problem catalog has that company and stage context.

Sort Colors

Capital One● MediumFULLTIMEONSITE INTERVIEW

Learn

Problem statement

You are given an integer array nums containing only 0, 1, and 2. Rearrange the array in place so that equal values are adjacent and the values appear in the order 0, 1, then 2.

Do not call a library sorting routine. Return the same array after rearranging it so the result can be evaluated.

Function

sortColors(nums: int[]) → int[]

Examples

Example 1

nums = [2,0,2,1,1,0]return = [0,0,1,1,2,2]

The two zeros come first, followed by the two ones and the two twos.

Example 2

nums = [2,0,1]return = [0,1,2]

Each color occurs once.

Constraints

1 <= nums.length <= 100000
nums[i] is 0, 1, or 2.
The rearrangement must use constant auxiliary space.