forked from TheAlgorithms/JavaScript
-
Notifications
You must be signed in to change notification settings - Fork 0
/
DutchNationalFlagSort.js
33 lines (31 loc) · 982 Bytes
/
DutchNationalFlagSort.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
/**
* @function dutchNationalFlagSort
* @description Dutch National Flag Sort is an algorithm to sort an array containing 0s, 1s, and 2s in linear time.
Time complexity of Dutch National Flag Sort Algorithm is O(n).
Auxiliary Space required for Dutch National Flag Sort Algorithm is O(1).
* @param {Integer[]} nums - Array of integers containing 0s, 1s, and 2s.
* @return {Integer[]} - Array of integers sorted in non-decreasing order.
* @see [Dutch National Flag Sort](https://en.wikipedia.org/wiki/Dutch_national_flag_problem)
*/
export function dutchNationalFlagSort (nums) {
let low = 0
let mid = 0
let high = nums.length - 1
while (mid <= high) {
switch (nums[mid]) {
case 0:
[nums[low], nums[mid]] = [nums[mid], nums[low]]
low++
mid++
break
case 1:
mid++
break
case 2:
[nums[mid], nums[high]] = [nums[high], nums[mid]]
high--
break
}
}
return nums
}