Preserve first max_element result

Co-authored-by: nunoplopes <2998477+nunoplopes@users.noreply.github.com>

Author: Copilot

rules/algorithm/tgt_refcount.rs

@@ -47,10 +47,11 @@ fn f8<T1: PartialOrd + Clone + ByteRepr>(a0: Ptr<T1>, a1: Ptr<T1>) -> Ptr<T1> {
     let count = a1.get_offset() - a0.get_offset();
     let max_index = PtrValueIter::new(&a0, count)
         .enumerate()
-        .max_by(|(_, val_a), (_, val_b)| {
+        .max_by(|(idx_a, val_a), (idx_b, val_b)| {
             val_a
                 .partial_cmp(val_b)
                 .unwrap_or(std::cmp::Ordering::Equal)
+                .then_with(|| idx_b.cmp(idx_a))
         })
         .map(|(idx, _)| idx)
         .unwrap_or(0);

rules/algorithm/tgt_unsafe.rs

@@ -49,7 +49,11 @@ unsafe fn f8<T1: PartialOrd>(a0: *mut T1, a1: *mut T1) -> *mut T1 {
     std::slice::from_raw_parts(a0, count)
         .iter()
         .enumerate()
-        .max_by(|(_, x), (_, y)| x.partial_cmp(y).unwrap_or(std::cmp::Ordering::Equal))
+        .max_by(|(idx_a, x), (idx_b, y)| {
+            x.partial_cmp(y)
+                .unwrap_or(std::cmp::Ordering::Equal)
+                .then_with(|| idx_b.cmp(idx_a))
+        })
         .map(|(i, _)| a0.add(i))
         .unwrap_or(a0)
 }

rules/vector/tgt_refcount.rs

@@ -152,7 +152,9 @@ fn f42(a0: Ptr<u8>, a1: Ptr<u8>) -> Ptr<u8> {
     let __count = a1.get_offset() - a0.get_offset();
     let max_index = PtrValueIter::new(&a0, __count)
         .enumerate()
-        .max_by_key(|&(_, val)| val)
+        .max_by(|&(idx_a, val_a), &(idx_b, val_b)| {
+            val_a.cmp(&val_b).then_with(|| idx_b.cmp(&idx_a))
+        })
         .map(|(idx, _)| idx)
         .unwrap_or(0);
 
@@ -312,7 +314,9 @@ fn f94(a0: Ptr<u8>, a1: Ptr<u8>) -> Ptr<u8> {
     let __count = a1.get_offset() - a0.get_offset();
     let max_index = PtrValueIter::new(&a0, __count)
         .enumerate()
-        .max_by_key(|&(_, val)| val)
+        .max_by(|&(idx_a, val_a), &(idx_b, val_b)| {
+            val_a.cmp(&val_b).then_with(|| idx_b.cmp(&idx_a))
+        })
         .map(|(idx, _)| idx)
         .unwrap_or(0);
 

rules/vector/tgt_unsafe.rs

@@ -184,7 +184,9 @@ unsafe fn f41<T1>(a0: &mut Vec<T1>) -> *const T1 {
 unsafe fn f42<T1: Ord>(a0: *const T1, a1: *const T1) -> *const T1 {
     core::slice::from_raw_parts(a0, (a1).offset_from(a0) as usize)
         .iter()
-        .max()
+        .enumerate()
+        .max_by(|(idx_a, val_a), (idx_b, val_b)| val_a.cmp(val_b).then_with(|| idx_b.cmp(idx_a)))
+        .map(|(_, value)| value)
         .unwrap()
 }
 
@@ -407,7 +409,9 @@ unsafe fn f93<T1>(a0: &mut Vec<T1>) -> *const T1 {
 unsafe fn f94<T1: Ord>(a0: *const T1, a1: *const T1) -> *const T1 {
     core::slice::from_raw_parts(a0, (a1).offset_from(a0) as usize)
         .iter()
-        .max()
+        .enumerate()
+        .max_by(|(idx_a, val_a), (idx_b, val_b)| val_a.cmp(val_b).then_with(|| idx_b.cmp(idx_a)))
+        .map(|(_, value)| value)
         .unwrap()
 }