#75 - Sort Colors

Medium • View on LeetCode • View on GitHub

Problem Description

Given an array nums with n objects colored red, white, or blue, sort them in-place so that objects of the same color are adjacent, with the colors in the order red, white, and blue.

We will use the integers 0, 1, and 2 to represent the color red, white, and blue, respectively.

You must solve this problem without using the library's sort function.

Solution

/**
 * @param {number[]} nums
 * @return {void} Do not return anything, modify nums in-place instead.
 */
var sortColors = function(nums) {
  const count = nums.reduce((a, n) => ++a[n] && a, [0, 0, 0]);
  for (let i = 0, j = 0; i < count.length; i++) {
    for (let k = 0; k < count[i]; k++) {
      nums[j++] = i;
    }
  }
  return nums;
};