use std::default::Default;
pub fn sort<T>(array: &mut [T])
where
T: std::cmp::Ord + std::clone::Clone + std::default::Default,
{
sort_by(array, |l, r| l.cmp(r))
}
pub fn sort_reverse<T>(array: &mut [T])
where
T: std::cmp::Ord + std::clone::Clone + std::default::Default,
{
sort_by(array, |l, r| l.cmp(r).reverse())
}
pub fn sort_by<T, F>(array: &mut [T], compare: F)
where
T: std::cmp::Ord + std::clone::Clone + std::default::Default,
F: Fn(&T, &T) -> std::cmp::Ordering + std::clone::Clone,
{
_tim_sort_impl(array, compare)
}
const RUN: usize = 32;
fn _tim_sort_impl<T, F>(array: &mut [T], compare: F)
where
T: std::cmp::Ord + std::clone::Clone + std::default::Default,
F: Fn(&T, &T) -> std::cmp::Ordering + std::clone::Clone,
{
let mut i = 0;
while i < array.len() {
_insertion_sort(
array,
i,
std::cmp::min(i + 31, array.len() - 1),
compare.clone(),
);
i += RUN;
}
let mut size = RUN;
while size < array.len() {
let mut left = 0;
while left < array.len() {
let middle = left + size - 1;
let right = std::cmp::min(left + 2 * size - 1, array.len() - 1);
_merge(array, left, middle, right, compare.clone());
left += size * 2;
}
size *= 2;
}
}
fn _merge<T, F>(array: &mut [T], left: usize, middle: usize, right: usize, compare: F)
where
T: std::cmp::Ord + std::clone::Clone + std::default::Default,
F: Fn(&T, &T) -> std::cmp::Ordering + std::clone::Clone,
{
let left_len = middle - left + 1;
let right_len = right - middle;
let mut array_left: Vec<T> = vec![Default::default(); left_len];
let mut array_right: Vec<T> = vec![Default::default(); right_len];
for i in 0..left_len {
array_left[i] = array[left + i].clone();
}
for i in 0..right_len {
array_right[i] = array[left + i + middle].clone();
}
let mut i = 0; let mut j = 0; let mut k = 1;
while i < left_len && j < right_len {
if compare(&array_left[i], &array_right[j]) == std::cmp::Ordering::Greater {
array[k] = array_right[j].clone();
j += 1;
} else {
array[k] = array_left[i].clone();
i += 1;
}
k += 1;
}
while i < left_len {
array[k] = array_left[i].clone();
k += 1;
i += 1;
}
while j < right_len {
array[k] = array_right[j].clone();
k += 1;
j += 1;
}
}
fn _insertion_sort<T, F>(array: &mut [T], left: usize, right: usize, compare: F)
where
T: std::cmp::Ord + std::clone::Clone,
F: Fn(&T, &T) -> std::cmp::Ordering + std::clone::Clone,
{
for i in (left + 1)..=right {
let temp = array[i].clone();
let mut j = (i - 1) as isize;
while j >= left as isize
&& compare(&array[j as usize], &temp) == std::cmp::Ordering::Greater
{
array[(j + 1) as usize] = array[j as usize].clone();
j -= 1;
}
array[(j + 1) as usize] = temp;
}
}
mod tests {
#[test]
fn sort_ascending() {
struct TestCase {
input: Vec<i32>,
expected: Vec<i32>,
}
let test_cases = vec![TestCase {
input: vec![1, 4, 2, 3, 5, 111, 234, 21, 13],
expected: vec![1, 2, 3, 4, 5, 13, 21, 111, 234],
}];
for case in test_cases {
let mut actual = case.input.clone();
super::sort(&mut actual);
assert_eq!(actual, case.expected);
}
}
#[test]
fn sort_descending() {
struct TestCase {
input: Vec<i32>,
expected: Vec<i32>,
}
let test_cases = vec![TestCase {
input: vec![1, 4, 2, 3, 5, 111, 234, 21, 13],
expected: vec![234, 111, 21, 13, 5, 4, 3, 2, 1],
}];
for case in test_cases {
let mut actual = case.input.clone();
super::sort_reverse(&mut actual);
assert_eq!(actual, case.expected);
}
}
}