-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
49 lines (43 loc) · 1.21 KB
/
index.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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
'use strict';
const assert = require('assert');
function swap(arr, i, j) {
const temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
function compare(a, b, arrayIsBigger) {
const flag = `${+Array.isArray(a)}${+Array.isArray(b)}`;
switch (flag) {
case '10':
return !arrayIsBigger;
case '01':
return arrayIsBigger;
default:
return a <= b;
}
}
function quickSort(arr, lo = 0, hi = arr.length, arrayIsBigger) {
function partition(lo, hi) {
swap(arr, lo, Math.round(Math.random() * (hi - lo) + lo));
const pivot = arr[lo];
while (lo < hi) {
while (lo < hi && compare(pivot, arr[hi], arrayIsBigger)) hi--;
arr[lo] = arr[hi];
while (lo < hi && compare(arr[lo], pivot, arrayIsBigger)) lo++;
arr[hi] = arr[lo];
}
arr[lo] = pivot;
return lo;
}
if (hi - lo < 2) return arr;
const mi = partition(lo, hi - 1);
quickSort(arr, lo, mi, arrayIsBigger);
quickSort(arr, mi + 1, hi, arrayIsBigger);
return arr;
}
function deepSort(arr, arrayIsBigger = true) {
assert(Array.isArray(arr));
arr.map(item => (Array.isArray(item) ? deepSort(item, arrayIsBigger) : item));
return quickSort(arr, 0, arr.length, arrayIsBigger);
}
module.exports = deepSort;