|
| 1 | +const swap = require("../工具/交换") |
| 2 | +const random = require("../工具/随机值") |
| 3 | + |
| 4 | +/** |
| 5 | + * 三路快速排序 |
| 6 | + * 将 arr[l...r] 分为 < v, === v, > v三部分 |
| 7 | + * 之后递归的对 < v, > v 两部分三路快排 |
| 8 | + * @param {number[]} arr |
| 9 | + */ |
| 10 | +function quickSort(arr) { |
| 11 | + _quickSort(arr, 0, arr.length - 1) |
| 12 | + return arr |
| 13 | +} |
| 14 | + |
| 15 | +/** |
| 16 | + * 对 arr[l...r] 部分进行快速排序 |
| 17 | + * @param {number[]} arr |
| 18 | + * @param {number} l 左边界 |
| 19 | + * @param {number} r 右边界 |
| 20 | + */ |
| 21 | +function _quickSort(arr, l, r) { |
| 22 | + if (l >= r) { |
| 23 | + return |
| 24 | + } |
| 25 | + let [p, q] = partition(arr, l, r) |
| 26 | + _quickSort(arr, l, p) |
| 27 | + _quickSort(arr, q, r) |
| 28 | +} |
| 29 | + |
| 30 | +/** |
| 31 | + * 对 arr[l...r] 部分进行快速排序 |
| 32 | + * @param {number[]} arr |
| 33 | + * @param {number} l 左边界 |
| 34 | + * @param {number} r 右边界 |
| 35 | + * @returns {number} 返回索引值p,使得arr[l...p-1] < arr[p] < arr[p+1...r] |
| 36 | + */ |
| 37 | +function partition(arr, left, right) { |
| 38 | + // 取一个基准值 取随机值 |
| 39 | + let rand = random(left, right) |
| 40 | + swap(arr, left, rand) |
| 41 | + let pivot = arr[left] |
| 42 | + |
| 43 | + // 三路 注意看注释里的区间 |
| 44 | + let lt = left // arr[left + 1...lt] < v |
| 45 | + let gt = right + 1 // arr[gt...r] > v |
| 46 | + let index = left + 1 // arr[lt + 1...index] === v |
| 47 | +} |
| 48 | + |
| 49 | +quickSort.sortName = "快速排序(三路)" |
| 50 | + |
| 51 | +module.exports = quickSort |
0 commit comments