quick-sort.js 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. /* -*- Mode: js; js-indent-level: 2; -*- */
  2. /*
  3. * Copyright 2011 Mozilla Foundation and contributors
  4. * Licensed under the New BSD license. See LICENSE or:
  5. * http://opensource.org/licenses/BSD-3-Clause
  6. */
  7. // It turns out that some (most?) JavaScript engines don't self-host
  8. // `Array.prototype.sort`. This makes sense because C++ will likely remain
  9. // faster than JS when doing raw CPU-intensive sorting. However, when using a
  10. // custom comparator function, calling back and forth between the VM's C++ and
  11. // JIT'd JS is rather slow *and* loses JIT type information, resulting in
  12. // worse generated code for the comparator function than would be optimal. In
  13. // fact, when sorting with a comparator, these costs outweigh the benefits of
  14. // sorting in C++. By using our own JS-implemented Quick Sort (below), we get
  15. // a ~3500ms mean speed-up in `bench/bench.html`.
  16. /**
  17. * Swap the elements indexed by `x` and `y` in the array `ary`.
  18. *
  19. * @param {Array} ary
  20. * The array.
  21. * @param {Number} x
  22. * The index of the first item.
  23. * @param {Number} y
  24. * The index of the second item.
  25. */
  26. function swap(ary, x, y) {
  27. var temp = ary[x];
  28. ary[x] = ary[y];
  29. ary[y] = temp;
  30. }
  31. /**
  32. * Returns a random integer within the range `low .. high` inclusive.
  33. *
  34. * @param {Number} low
  35. * The lower bound on the range.
  36. * @param {Number} high
  37. * The upper bound on the range.
  38. */
  39. function randomIntInRange(low, high) {
  40. return Math.round(low + (Math.random() * (high - low)));
  41. }
  42. /**
  43. * The Quick Sort algorithm.
  44. *
  45. * @param {Array} ary
  46. * An array to sort.
  47. * @param {function} comparator
  48. * Function to use to compare two items.
  49. * @param {Number} p
  50. * Start index of the array
  51. * @param {Number} r
  52. * End index of the array
  53. */
  54. function doQuickSort(ary, comparator, p, r) {
  55. // If our lower bound is less than our upper bound, we (1) partition the
  56. // array into two pieces and (2) recurse on each half. If it is not, this is
  57. // the empty array and our base case.
  58. if (p < r) {
  59. // (1) Partitioning.
  60. //
  61. // The partitioning chooses a pivot between `p` and `r` and moves all
  62. // elements that are less than or equal to the pivot to the before it, and
  63. // all the elements that are greater than it after it. The effect is that
  64. // once partition is done, the pivot is in the exact place it will be when
  65. // the array is put in sorted order, and it will not need to be moved
  66. // again. This runs in O(n) time.
  67. // Always choose a random pivot so that an input array which is reverse
  68. // sorted does not cause O(n^2) running time.
  69. var pivotIndex = randomIntInRange(p, r);
  70. var i = p - 1;
  71. swap(ary, pivotIndex, r);
  72. var pivot = ary[r];
  73. // Immediately after `j` is incremented in this loop, the following hold
  74. // true:
  75. //
  76. // * Every element in `ary[p .. i]` is less than or equal to the pivot.
  77. //
  78. // * Every element in `ary[i+1 .. j-1]` is greater than the pivot.
  79. for (var j = p; j < r; j++) {
  80. if (comparator(ary[j], pivot) <= 0) {
  81. i += 1;
  82. swap(ary, i, j);
  83. }
  84. }
  85. swap(ary, i + 1, j);
  86. var q = i + 1;
  87. // (2) Recurse on each half.
  88. doQuickSort(ary, comparator, p, q - 1);
  89. doQuickSort(ary, comparator, q + 1, r);
  90. }
  91. }
  92. /**
  93. * Sort the given array in-place with the given comparator function.
  94. *
  95. * @param {Array} ary
  96. * An array to sort.
  97. * @param {function} comparator
  98. * Function to use to compare two items.
  99. */
  100. exports.quickSort = function (ary, comparator) {
  101. doQuickSort(ary, comparator, 0, ary.length - 1);
  102. };