我在上一讲为你介绍了用 JS 实现数组的各种排序,那么这一讲作为进阶,我要带你了解一下 JS 提供的数组排序的 sort 方法。数组排序在日常工作中经常会出现,除了上一讲介绍的排序算法外,通过 sort 方法也可以实现数组的排序,因此这一讲我会带你剖析 JS 数组 sort 方法的底层实现。

我们比较常用的是直接调用 JS 的 sort 方法,这一讲会围绕原生 JS 方法,并结合 V8 排序的代码一起来分析,以便你碰到类似的 JS 排序问题时能够轻松应对。

那么,在课程开始前请你先思考一下:

  1. sort 方法到底是用了哪种排序思路?
  2. sort 方法里面的参数对比函数是什么意思?

带着问题,我们开始说说 sort 方法的基本内容。

sort 方法的基本使用

sort 方法是对数组元素进行排序,默认排序顺序是先将元素转换为字符串,然后再进行排序,先来看一下它的语法:

arr.sort([compareFunction])

其中 compareFunction 用来指定按某种顺序进行排列的函数,如果省略不写,元素按照转换为字符串的各个字符的 Unicode 位点进行排序。我们来看一下代码。

  1. const months = ['March', 'Jan', 'Feb', 'Dec'];
  2. months.sort();
  3. console.log(months);
  4. const array1 = [1, 30, 4, 21, 100000];
  5. array1.sort();
  6. console.log(array1);

从上面的执行结果可以看出,如果不加参数,在第二段代码中,21 会排到 4 的前面。这样按照从小到大的逻辑是行不通的,如果想要按照从小到大排序或者从大到小排序,那么上面的代码就需要调整为下面这样。

  1. const array1 = [1, 30, 4, 21, 100000];
  2. array1.sort((a,b) => b - a);
  3. console.log(array1);
  4. const array1 = [1, 30, 4, 21, 100000];
  5. array1.sort((a,b) => a - b);
  6. console.log(array1);

如果指明了 compareFunction 参数 ,那么数组会按照调用该函数的返回值排序,即 a 和 b 是两个将要被比较的元素:

  • 如果 compareFunction(a, b)小于 0,那么 a 会被排列到 b 之前;
  • 如果 compareFunction(a, b)等于 0,a 和 b 的相对位置不变;
  • 如果 compareFunction(a, b)大于 0,b 会被排列到 a 之前。

说完 sort 方法的对比函数,下面我们来看一下 sort 的底层实现。

sort 方法的底层实现

相信你对 JS 数组的 sort 方法已经不陌生了,上面我也对它的用法进行了详细的介绍。那么它的内部是如何实现的呢?如果你能够进入它的内部看一看源码,理解背后的设计,这对编程思维的提升是一个很好的帮助。

sort 方法在 V8 内部相较于其他方法而言是一个比较难的算法,对于很多边界情况结合排序算法做了反复的优化,但是这里我不会直接拿源码来讲,而是会根据源码的思路,循序善诱地带你实现一个跟引擎性能类似的排序算法,并且一步步拆解其中的奥秘。

这里你需要回顾上一讲说的插入排序和快速排序,接下来我们就对 sort 源码进行分析。

底层 sort 源码分析

先大概来梳理一下源码中排序的思路(下面的源码均来自 V8 源码中关于 sort 排序的摘要,地址:V8 源码 sort 排序部分)。

通过研究源码我们先直接看一下结论,如果要排序的元素个数是 n 的时候,那么就会有以下几种情况:

  1. 当 n<=10 时,采用插入排序;
  2. 当 n>10 时,采用三路快速排序;
  3. 10<n <=1000,采用中位数作为哨兵元素;
  4. n>1000,每隔 200~215 个元素挑出一个元素,放到一个新数组中,然后对它排序,找到中间位置的数,以此作为中位数。

在得出这个结论之前,我觉得有必要让你了解为什么这么做。下面就一起来思考两个问题。

1. 为什么元素个数少的时候要采用插入排序?

虽然插入排序理论上是平均时间复杂度为 O(n^2) 的算法,快速排序是一个平均 O(nlogn) 级别的算法。但是别忘了,这只是理论上平均的时间复杂度估算,但是它们也有最好的时间复杂度情况,而插入排序在最好的情况下时间复杂度是 O(n)。

在实际情况中两者的算法复杂度前面都会有一个系数,当 n 足够小的时候,快速排序 nlogn 的优势会越来越小。倘若插入排序的 n 足够小,那么就会超过快排。而事实上正是如此,插入排序经过优化以后,对于小数据集的排序会有非常优越的性能,很多时候甚至会超过快排。因此,对于很小的数据量,应用插入排序是一个非常不错的选择。

2. 为什么要花这么大的力气选择哨兵元素?

因为快速排序的性能瓶颈在于递归的深度,最坏的情况是每次的哨兵都是最小元素或者最大元素,那么进行 partition(一边是小于哨兵的元素,另一边是大于哨兵的元素)时,就会有一边是空的。如果这么排下去,递归的层数就达到了 n , 而每一层的复杂度是 O(n),因此快排这时候会退化成 O(n^2) 级别。

这种情况是要尽力避免的,那么如何来避免?就是让哨兵元素尽可能地处于数组的中间位置,让最大或者最小的情况尽可能少。这时候,你就能理解 V8 里面所做的各种优化了。

接下来,我们看一下官方实现的 sort 排序算法的代码基本结构。

  1. function ArraySort(comparefn) {
  2. CHECK_OBJECT_COERCIBLE(this,"Array.prototype.sort");
  3. var array = TO_OBJECT(this);
  4. var length = TO_LENGTH(array.length);
  5. return InnerArraySort(array, length, comparefn);
  6. }
  7. function InnerArraySort(array, length, comparefn) {
  8. if (!IS_CALLABLE(comparefn)) {
  9. comparefn = function (x, y) {
  10. if (x === y) return 0;
  11. if (%_IsSmi(x) && %_IsSmi(y)) {
  12. return %SmiLexicographicCompare(x, y);
  13. }
  14. x = TO_STRING(x);
  15. y = TO_STRING(y);
  16. if (x == y) return 0;
  17. else return x < y ? -1 : 1;
  18. };
  19. }
  20. function InsertionSort(a, from, to) {
  21. for (var i = from + 1; i < to; i++) {
  22. var element = a[i];
  23. for (var j = i - 1; j >= from; j--) {
  24. var tmp = a[j];
  25. var order = comparefn(tmp, element);
  26. if (order > 0) {
  27. a[j + 1] = tmp;
  28. } else {
  29. break;
  30. }
  31. }
  32. a[j + 1] = element;
  33. }
  34. }
  35. function GetThirdIndex(a, from, to) {
  36. var t_array = new InternalArray();
  37. var increment = 200 + ((to - from) & 15);
  38. var j = 0;
  39. from += 1;
  40. to -= 1;
  41. for (var i = from; i < to; i += increment) {
  42. t_array[j] = [i, a[i]];
  43. j++;
  44. }
  45. t_array.sort(function(a, b) {
  46. return comparefn(a[1], b[1]);
  47. });
  48. var third_index = t_array[t_array.length >> 1][0];
  49. return third_index;
  50. }
  51. function QuickSort(a, from, to) {
  52. var third_index = 0;
  53. while (true) {
  54. if (to - from <= 10) {
  55. InsertionSort(a, from, to);
  56. return;
  57. }
  58. if (to - from > 1000) {
  59. third_index = GetThirdIndex(a, from, to);
  60. } else {
  61. third_index = from + ((to - from) >> 1);
  62. }
  63. var v0 = a[from];
  64. var v1 = a[to - 1];
  65. var v2 = a[third_index];
  66. var c01 = comparefn(v0, v1);
  67. if (c01 > 0) {
  68. var tmp = v0;
  69. v0 = v1;
  70. v1 = tmp;
  71. }
  72. var c02 = comparefn(v0, v2);
  73. if (c02 >= 0) {
  74. var tmp = v0;
  75. v0 = v2;
  76. v2 = v1;
  77. v1 = tmp;
  78. } else {
  79. var c12 = comparefn(v1, v2);
  80. if (c12 > 0) {
  81. var tmp = v1;
  82. v1 = v2;
  83. v2 = tmp;
  84. }
  85. }
  86. a[from] = v0;
  87. a[to - 1] = v2;
  88. var pivot = v1;
  89. var low_end = from + 1;
  90. var high_start = to - 1;
  91. a[third_index] = a[low_end];
  92. a[low_end] = pivot;
  93. partition: for (var i = low_end + 1; i < high_start; i++) {
  94. var element = a[i];
  95. var order = comparefn(element, pivot);
  96. if (order < 0) {
  97. a[i] = a[low_end];
  98. a[low_end] = element;
  99. low_end++;
  100. } else if (order > 0) {
  101. do {
  102. high_start--;
  103. if (high_start == i) break partition;
  104. var top_elem = a[high_start];
  105. order = comparefn(top_elem, pivot);
  106. } while (order > 0);
  107. a[i] = a[high_start];
  108. a[high_start] = element;
  109. if (order < 0) {
  110. element = a[i];
  111. a[i] = a[low_end];
  112. a[low_end] = element;
  113. low_end++;
  114. }
  115. }
  116. }
  117. if (to - high_start < low_end - from) {
  118. QuickSort(a, high_start, to);
  119. to = low_end;
  120. } else {
  121. QuickSort(a, from, low_end);
  122. from = high_start;
  123. }
  124. }
  125. }

从上面的源码分析来看,当数据量小于 10 的时候用插入排序;当数据量大于 10 之后采用三路快排;当数据量为 10~1000 时候直接采用中位数为哨兵元素;当数据量大于 1000 的时候就开始寻找哨兵元素。

我们直接从上面的源码中就可以看到整个 sort 源码的编写逻辑,也就是上面总结分析的逻辑对应实现。如果你还是没有理解得很好,我建议你再重新看一下插入排序和快速排序的核心逻辑。其实关键点在于根据数据量的大小,从而确定用什么排序来解决;时间复杂度是根据数据量的大小,从而进行变化的,这一点需要深入理解。

总结

那么关于 sort 排序方法我就介绍到这里了。这一讲,我把 JS 中的 sort 方法详细讲解了一遍,同时又带你剖析了 sort 方法中内部的原理实现,你需要好好学习排序相关的算法,才能真正理解 V8 源码中实现的排序逻辑。

此外,关于排序时间复杂度也不用死记硬背,在不同的数据量情况下,不代表某种排序一定就要比另外一种排序速度快,这点你要牢记,然后根据不同的场景进行不同的分析。

那么讲到这里,我们再来整理一下快速排序和插入排序的最好以及最快情况下的时间复杂度的对比,请看下面的表格。

|11 | 数组排序(下):sort 排序方法的实现原理 - 图1

将这两个排序的时间复杂度对比来看,如果当 n 足够小的时候,最好的情况下,插入排序的时间复杂度为 O(n) 要优于快速排序的 O(nlogn),因此就解释了这里当 V8 实现 JS 数组排序算法时,数据量较小的时候会采用插入排序的原因了。

在日常的前端开发工作中,对此的应用会比较多,研究源码的机会也是相对较少的。通过本讲的学习,我还是希望你能够多想想日常工作中经常用到的 JS 方法,及其底层源代码的实现逻辑,从而整体提升 JS 的编程能力和理解能力。

这一讲我只是带你剖析了 sort 的源码实现,下一讲我会带你手写 JS 数组多个方法的底层实现。

另外在后面的课程中,类似这样的源码剖析还会有很多,在看每一个部分的同时,希望你能多练习、多研究,也欢迎你在下方留言发表自己在学习过程中遇到的困惑以及学习感悟等,让我们共同进步。