[libc++] Optimize make_heap() and sift_down()

Author: nukyan

libcxx/include/__algorithm/make_heap.h

@@ -34,10 +34,12 @@ __make_heap(_RandomAccessIterator __first, _RandomAccessIterator __last, _Compar
   using difference_type = typename iterator_traits<_RandomAccessIterator>::difference_type;
   difference_type __n   = __last - __first;
   if (__n > 1) {
-    // start from the first parent, there is no need to consider children
-    for (difference_type __start = (__n - 2) / 2; __start >= 0; --__start) {
+    difference_type __start = __n / 2;
+    do {
+      // start from the first parent, there is no need to consider children
+      --__start;
       std::__sift_down<_AlgPolicy>(__first, __comp_ref, __n, __first + __start);
-    }
+    } while (__start != 0);
   }
 }
 

libcxx/include/__algorithm/sift_down.h

@@ -44,10 +44,13 @@ __sift_down(_RandomAccessIterator __first,
   __child                         = 2 * __child + 1;
   _RandomAccessIterator __child_i = __first + __child;
 
-  if ((__child + 1) < __len && __comp(*__child_i, *(__child_i + difference_type(1)))) {
-    // right-child exists and is greater than left-child
-    ++__child_i;
-    ++__child;
+  if ((__child + 1) < __len) {
+    _RandomAccessIterator __right_child_i = _Ops::next(__child_i);
+    if (__comp(*__child_i, *__right_child_i)) {
+      // right-child exists and is greater than left-child
+      __child_i = __right_child_i;
+      ++__child;
+    }
   }
 
   // check if we are in heap-order
@@ -61,17 +64,20 @@ __sift_down(_RandomAccessIterator __first,
     *__start = _Ops::__iter_move(__child_i);
     __start  = __child_i;
 
-    if ((__len - 2) / 2 < __child)
+    __child = 2 * __child + 1;
+    if (!(__child < __len))
       break;
 
     // recompute the child based off of the updated parent
-    __child   = 2 * __child + 1;
     __child_i = __first + __child;
 
-    if ((__child + 1) < __len && __comp(*__child_i, *(__child_i + difference_type(1)))) {
-      // right-child exists and is greater than left-child
-      ++__child_i;
-      ++__child;
+    if ((__child + 1) < __len) {
+      _RandomAccessIterator __right_child_i = _Ops::next(__child_i);
+      if (__comp(*__child_i, *__right_child_i)) {
+        // right-child exists and is greater than left-child
+        __child_i = __right_child_i;
+        ++__child;
+      }
     }
 
     // check if we are in heap-order