Sort Colors

mediumcodingAlgorithmsTwo PointersArraysSorting~25 min

Asked at Google, Meta

Overview

Given an array nums containing only 0, 1, and 2, sort it in-place so that objects of the same color are adjacent (0s, then 1s, then 2s). Do not use the library sort function.

This is the Dutch National Flag problem.

Constraints

1 <= nums.length <= 300
nums[i] is 0, 1, or 2.
One-pass with O(1) extra space.

Examples

sortColors([2,0,2,1,1,0]);
// => [0,0,1,1,2,2]

sortColors([2,0,1]);
// => [0,1,2]

Solution

Reveal solution

function sortColors(nums) {
  let lo = 0, mid = 0, hi = nums.length - 1;
  while (mid <= hi) {
    if (nums[mid] === 0) { [nums[lo], nums[mid]] = [nums[mid], nums[lo]]; lo++; mid++; }
    else if (nums[mid] === 1) mid++;
    else { [nums[mid], nums[hi]] = [nums[hi], nums[mid]]; hi--; }
  }
  return nums;
}

Resources

sort-colors.js

Sort Colors

mediumcodingAlgorithmsTwo Pointers

Overview

Given an array nums containing only 0, 1, and 2, sort it in-place so that objects of the same color are adjacent (0s, then 1s, then 2s). Do not use the library sort function.

This is the Dutch National Flag problem.

Constraints

1 <= nums.length <= 300
nums[i] is 0, 1, or 2.
One-pass with O(1) extra space.

Examples

sortColors([2,0,2,1,1,0]);
// => [0,0,1,1,2,2]

sortColors([2,0,1]);
// => [0,1,2]

Solution

Reveal solution

function sortColors(nums) {
  let lo = 0, mid = 0, hi = nums.length - 1;
  while (mid <= hi) {
    if (nums[mid] === 0) { [nums[lo], nums[mid]] = [nums[mid], nums[lo]]; lo++; mid++; }
    else if (nums[mid] === 1) mid++;
    else { [nums[mid], nums[hi]] = [nums[hi], nums[mid]]; hi--; }
  }
  return nums;
}

Resources

NameTopicDifficulty

Add BinaryAlgorithmseasy Balanced Binary TreeAlgorithmseasy Balanced BracketsAlgorithmseasy Binary SearchAlgorithmseasy Binary Search Tree Lowest Common AncestorAlgorithmseasy Binary Tree Maximum DepthAlgorithmseasy Binary Tree SubtreeAlgorithmseasy Bit CountingAlgorithmseasy Bit ReversalAlgorithmseasy Count Set Bits in a Binary NumberAlgorithmseasy Diameter of Binary TreeAlgorithmseasy Find Duplicates in ArrayAlgorithmseasy First Bad VersionAlgorithmseasy Flip Binary TreeAlgorithmseasy Flood FillAlgorithmseasy Linked List Detect CycleAlgorithmseasy Linked List ReversalAlgorithmseasy Linked Lists Combine Two SortedAlgorithmseasy Longest PalindromeAlgorithmseasy Middle of the Linked ListAlgorithmseasy Most Common ElementsAlgorithmseasy Optimal Stock TradingAlgorithmseasy Pair SumAlgorithmseasy Ransom NoteAlgorithmseasy Staircase Climbing CombinationsAlgorithmseasy String AnagramAlgorithmseasy String PalindromeAlgorithmseasy 01 MatrixAlgorithmsmedium Accounts MergeAlgorithmsmedium Array Product Excluding CurrentAlgorithmsmedium Binary Search Tree Kth Smallest ElementAlgorithmsmedium Binary Tree Level Order TraversalAlgorithmsmedium Binary Tree Rebuilding from Preorder and Inorder TraversalsAlgorithmsmedium Binary Tree Right Side ViewAlgorithmsmedium Combinations for Target SumAlgorithmsmedium Count Islands in a GridAlgorithmsmedium Course DependencyAlgorithmsmedium Decode MessageAlgorithmsmedium Disjoint IntervalsAlgorithmsmedium Distinct Paths in GridAlgorithmsmedium Evaluate Reverse Polish NotationAlgorithmsmedium Find All Anagrams in a StringAlgorithmsmedium Find Element in Rotated ArrayAlgorithmsmedium Find the Longest Palindromic SubstringAlgorithmsmedium Find Word in GridAlgorithmsmedium Graph CloneAlgorithmsmedium Graph Count Connected ComponentsAlgorithmsmedium Is the Graph a TreeAlgorithmsmedium K Closest Points to OriginAlgorithmsmedium Letter Combinations of a Phone NumberAlgorithmsmedium Longest Increasing SubsequenceAlgorithmsmedium Longest Non-repeating SubstringAlgorithmsmedium Longest Repeating Substring After ReplacementsAlgorithmsmedium Lowest Common Ancestor of a Binary TreeAlgorithmsmedium Matrix RotationAlgorithmsmedium Matrix Spiral TraversalAlgorithmsmedium Matrix ZeroingAlgorithmsmedium Maximum Sum in Contiguous ArrayAlgorithmsmedium Maximum Water Trapped Between WallsAlgorithmsmedium Merge New IntervalAlgorithmsmedium Merge Overlapping IntervalsAlgorithmsmedium Minimum Coins for ChangeAlgorithmsmedium Minimum Height TreesAlgorithmsmedium Palindromic SubstringsAlgorithmsmedium Partition Equal Subset SumAlgorithmsmedium PermutationsAlgorithmsmedium Rotting OrangesAlgorithmsmedium Segment WordsAlgorithmsmedium Sort ColorsAlgorithmsmedium String Anagram GroupsAlgorithmsmedium String to Integer (atoi)Algorithmsmedium SubsetsAlgorithmsmedium Sum Without AdditionAlgorithmsmedium Task CoordinationAlgorithmsmedium Trie (Prefix Tree)Algorithmsmedium Triplet SumAlgorithmsmedium Validate Binary Search TreeAlgorithmsmedium Basic CalculatorAlgorithmshard Binary Tree EqualAlgorithmshard Binary Tree Maximum Total PathAlgorithmshard Delete Nth Node from End of Linked ListAlgorithmshard End of Array ReachableAlgorithmshard Extraterrestrial LanguageAlgorithmshard Find Missing Number in SequenceAlgorithmshard Find Words in GridAlgorithmshard Largest Rectangle in HistogramAlgorithmshard Linked Lists Combine K SortedAlgorithmshard Longest Common SubsequenceAlgorithmshard Longest Consecutive Number SequenceAlgorithmshard Maximum Product in Contiguous ArrayAlgorithmshard Maximum Profit in Job SchedulingAlgorithmshard Meeting CalendarAlgorithmshard Minimum Meeting Rooms NeededAlgorithmshard Neighborhood TheftAlgorithmshard Neighborhood Theft (Circular)Algorithmshard Number Stream MedianAlgorithmshard Ocean FlowAlgorithmshard Rearrange Linked ListAlgorithmshard Shortest Substring Containing CharactersAlgorithmshard Smallest Element in Rotated Sorted ArrayAlgorithmshard Trapping Rain WaterAlgorithmshard Word FinderAlgorithmshard Word LadderAlgorithmshard

103 of 103 problems