]> git.lizzy.rs Git - rust.git/commitdiff
sort: Fast path for already sorted data
authorUlrik Sverdrup <bluss@users.noreply.github.com>
Sat, 7 Nov 2015 16:39:36 +0000 (17:39 +0100)
committerUlrik Sverdrup <bluss@users.noreply.github.com>
Sat, 7 Nov 2015 16:45:14 +0000 (17:45 +0100)
When merging two sorted blocks `left` and `right` if the last element in
`left` is <= the first in `right`, the blocks are already sorted.

Add this as an additional fast path by simply copying the whole left
block into the output and advancing the left pointer. The right block is
then treated the same way by the already present logic in the merge
loop.

Reduces runtime of .sort() to less than 50% of the previous, if the data
was already perfectly sorted. Sorted data with a few swaps are also
sorted quicker than before. The overhead of one comparison per merge
seems to be negligible.

src/libcollections/slice.rs

index ea4830fc3e6cefab41c11700a2e8e924a9637007..cb3f39e0cac4b2c187da438cbd6dccd8c2895964 100644 (file)
@@ -1066,6 +1066,16 @@ fn merge_sort<T, F>(v: &mut [T], mut compare: F) where F: FnMut(&T, &T) -> Order
                 let mut out = buf_tmp.offset(start as isize);
                 let out_end = buf_tmp.offset(right_end_idx as isize);
 
+                // if left[last] <= right[0], they are already in order:
+                // fast-forward the left side (the right side is handled
+                // in the loop).
+                if compare(&*right.offset(-1), &*right) != Greater {
+                    let elems = (right_start as usize - left as usize) / mem::size_of::<T>();
+                    ptr::copy_nonoverlapping(&*left, out, elems);
+                    out = out.offset(elems as isize);
+                    left = right_start;
+                }
+
                 while out < out_end {
                     // Either the left or the right run are exhausted,
                     // so just copy the remainder from the other run