/*
//利用STL库函数,引入了额外空间
void heapsort(vector<int> &arr) {
priority_queue<int> pq{ arr.begin(), arr.end() };
for (int i = arr.size() - 1; i >= 0; i--) {
arr[i] = pq.top();
pq.pop();
}
}
//利用STL库函数
void heapsort(vector<int> &arr) {
make_heap(arr.begin(), arr.end());
for (int i = arr.size() - 1; i >= 0; i--) {
swap(arr[0], arr[i]);
make_heap(arr.begin(), arr.begin() + i);
}
}
*/
//自己实现堆结构
void adjust_heap(vector<int> &arr, int i, int size) {//i代表当前节点的下标,此函数就是对此节点作为根节点的子树进行调整,使之满足堆的规律
int l_child = i * 2 + 1;//左孩子节点下标
int r_child = i * 2 + 2;//右孩子节点下标
int max = i;
//max为最大节点的下标,最大节点从节点i与他的孩子中选
if (l_child < size && arr[l_child] > arr[max]) max = l_child;
if (r_child < size && arr[r_child] > arr[max]) max = r_child;
//如果max指的不是节点i,说明需要把最大的节点换到根节点,然而这样就破环他的孩子作为根节点的排列规律,因此还需要递归对他的孩子进行处理
if (max != i) {
swap(arr[max], arr[i]);
adjust_heap(arr, max, size);
}
}
void heapsort(vector<int> &arr) {
//通过这个for循环,可以使当前数组排列成最大堆的堆结构
for (int i = arr.size() / 2 - 1; i >= 0; i--) {
adjust_heap(arr, i, arr.size()); //为什么下标要从(n/2)-1开始,其中n是数组大小,因为这是最后一个叶子节点的父节点下标
}
//提出最大的节点把它放到数组最后,之后在进行一下堆结构调整,不断重复这一过程,相当于由大到小从后往前排列
for (int i = arr.size() - 1; i >= 0; i--) {
swap(arr[0], arr[i]);
adjust_heap(arr, 0, i);
}
}