Sort Colors

Medium

Given an array nums with n objects colored red, white, or blue (0, 1, 2), sort them in-place so objects of the same color are adjacent, in the order 0, 1, 2. Return the sorted array. Use the one-pass Dutch National Flag algorithm.

Example 1

Input:  nums = [2,0,2,1,1,0]
Output: [0,0,1,1,2,2]

Example 2

Input:  nums = [2,0,1]
Output: [0,1,2]

Constraints

• n == nums.length
• 1 <= n <= 300
• nums[i] is 0, 1, or 2.