search.js: introduce optimized removeIdxListAsc routine

Author: lolbinarycat

src/librustdoc/html/static/js/search.js

@@ -1076,6 +1076,34 @@ function isPathSeparator(c) {
     return c === ":" || c === " ";
 }
 
+/**
+ * Given an array and an ascending list of indices,
+ * efficiently removes each index in the array.
+ *
+ * @template T
+ * @param {Array<T>} a
+ * @param {Array<number>} idxList
+ */
+function removeIdxListAsc(a, idxList) {
+    if (idxList.length === 0) {
+        return;
+    }
+    let removed = 0;
+    let i = idxList[0];
+    let nextToRemove = idxList[0];
+    while (i < a.length - idxList.length) {
+        while (i === nextToRemove && removed < idxList.length) {
+            removed++;
+            i++;
+            nextToRemove = idxList[removed];
+        }
+        a[i] = a[i + removed];
+        i++;
+    }
+    // truncate array
+    a.length -= idxList.length;
+}
+
 /**
  * @template T
  */
@@ -2616,7 +2644,7 @@ class DocSearch {
          */
         const transformResults = (results, typeInfo, duplicates) => {
             /** @type {rustdoc.ResultObject[]} */
-            let out = [];
+            const out = [];
 
             // if we match a trait-associated item, we want to go back and
             // remove all the items that are their equivalent but in an impl block.
@@ -2713,16 +2741,9 @@ class DocSearch {
                         list.push(out.length);
                         traitImplIdxMap.set(obj.traitPath, list);
                     } else {
-                        // FIXME: this is `O(n*m)` because we're repeatedly
-                        // shifting with Array.splice, but could be `O(n+m)` if
-                        // we did the shifting manually in a more clever way.
                         const toRemove = traitImplIdxMap.get(obj.fullPath);
                         if (toRemove) {
-                            // iterate in reverse order so we don't shift the indexes
-                            for (let i = toRemove.length - 1; i >= 0; i--) {
-                                const rmIdx = toRemove[i];
-                                out = out.splice(rmIdx, 1);
-                            }
+                            removeIdxListAsc(out, toRemove);
                         }
                         traitImplIdxMap.delete(obj.fullPath);
                     }