rust-algorithms:8-堆排序

godme發表於2022-07-25
RustGo排序
系列文章：
rust-algorithms:1-插入排序 rust-algorithms:2-氣泡排序 rust-algorithms:3-桶排序 rust-algorithms:4-雞尾酒排序 rust-algorithms:5-梳排序 rust-algorithms:6-計數排序 rust-algorithms:7-地精排序 rust-algorithms:8-堆排序 rust-algorithms:9-歸併排序 rust-algorithms:10-奇偶排序 rust-algorithms:11-快速排序 rust-algorithms:12-基數排序 rust-algorithms:13-選擇排序 rust-algorithms:14-希爾排序 rust-algorithms:15-臭皮匠排序 rust-algorithms:16-字元逆序 rust-algorithms:17-BWT 

pub fn heap_sort<T: PartialOrd>(arr: &mut [T]) {
    let len = arr.len();
    // 建堆，從尾到首
    for index in (0..len / 2).rev() {
        heapify(arr, index, len);
    }
    // 收縮堆，從尾到首
    for index in (1..len).rev() {
        // 1. 最大的移到末尾
        // 2. 重建堆(收縮)
        arr.swap(0, index);
        heapify(arr, 0, index);
    }

}

fn heapify<T: PartialOrd>(arr: &mut [T], root: usize, end: usize) {
    let mut largest_index = root;
    // 找到最大值
    let left_index = 2 * root + 1;
    if left_index < end && arr[left_index] > arr[largest_index] {
        largest_index = left_index;
    }
    let right_index = left_index + 1;
    if right_index < end && arr[right_index] > arr[largest_index] {
        largest_index = right_index;
    }
    // 交換最大值
    if largest_index != root {
        arr.swap(largest_index, root);
        heapify(arr, largest_index, end)
    }
}
本作品採用《CC 協議》，轉載必須註明作者和本文連結

相關文章