解法:

20240510_193050

最后一个非叶子节点就是最后一个节点的父节点

进行一次最小堆调整(如视频)

#include<iostream>
#include<vector>
using namespace std;
void min_heapfy(vector<int>& a,int sta,int end) {
	int dad = sta;
	int son = 2 * sta + 1;
	while (son < end) {
		if (son + 1 < end && a[son] > a[son + 1])
			son++;
		if (a[dad] < a[son]) 
			return;
		else {
			swap(a[dad], a[son]);
			dad = son;
			son = 2 * dad + 1;
		}
	}
}
int main() {
	int n;
	cin >> n;
	vector<int> vec(n);
	for (int& x : vec) cin >> x;
	for (int i = n / 2 - 1; ~i; i--) {
		min_heapfy(vec, i, n);
	}
	for (int x : vec) cout << x << " ";
	return 0;
}

要实现堆排序也非常简单,只需要按照如下步骤:

已知一次最大堆调整得到arr[0](堆顶)最小

1.堆顶和堆底arr[n-1]交换

2.arr[0]到arr[n-2]进行最大堆调整

3.重复至剩下最后一个元素

for (int i = n - 1; i > 0; i--) {
	swap(vec[0], vec[i]);
	min_heapfy(vec, 0, i);
}

得到降序

更多推荐