xmengnet/hello-algo
1
0
Fork 0
mirror of https://github.com/krahets/hello-algo.git synced 2024-12-25 01:06:30 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix panic when array's len is less than 2 (#1353)

Browse source
This commit is contained in:
rongyi 2024-05-11 17:59:11 +08:00 committed by GitHub
parent 1f606d6852
commit 7a96f6a743
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
codes/rust/chapter_sorting/heap_sort.rs
View file

@ -35,11 +35,11 @@ fn sift_down(nums: &mut [i32], n: usize, mut i: usize) {
/* 堆排序 */ /* 堆排序 */
fn heap_sort(nums: &mut [i32]) { fn heap_sort(nums: &mut [i32]) {
// 建堆操作：堆化除叶节点以外的其他所有节点 // 建堆操作：堆化除叶节点以外的其他所有节点
for i in (0..=nums.len() / 2 - 1).rev() { for i in (0..nums.len() / 2).rev() {
sift_down(nums, nums.len(), i); sift_down(nums, nums.len(), i);
} }
// 从堆中提取最大元素，循环 n-1 轮 // 从堆中提取最大元素，循环 n-1 轮
for i in (1..=nums.len() - 1).rev() { for i in (1..nums.len()).rev() {
// 交换根节点与最右叶节点（交换首元素与尾元素） // 交换根节点与最右叶节点（交换首元素与尾元素）
let tmp = nums[0]; let tmp = nums[0];
nums[0] = nums[i]; nums[0] = nums[i];