父节点的索引为i,lchid的索引为2 * i + 1,rchild的索引为2 * i + 2
最后一个含有孩子节点的父节点索引为:(v.size() - 1) / 2
堆排序分3步:
① 维护堆的性质
② 建堆(大根堆,升序排;小根堆,降序排)
③ 排序
#include <iostream>
#include <vector>
#include <algorithm>
using std::vector;
using std::cout;
using std::swap;
// 维护大根堆结点的属性【n为要维护根的大小的索引,i为要调整的节点】
void maxHeapify(vector<int>& v,int i,int n)
{
int maxIdx = i;
int lchildIdx = 2 * i + 1;
int rchildIdx = 2 * i + 2;
if(lchildIdx <= n && v[maxIdx] < v[lchildIdx])
maxIdx = lchildIdx;
if(rchildIdx <= n && v[maxIdx] < v[rchildIdx])
maxIdx = rchildIdx;
if(maxIdx != i)
{
swap(v[maxIdx],v[i]);
maxHeapify(v,maxIdx,n);
}
}
// 建立大根堆
void buildMaxSort(vector<int>& v)
{
for(int i = (v.size() - 1) / 2;i >= 0;-- i)
{
// 维护结点的属性
maxHeapify(v,i,v.size() - 1); // 这里传入最后一个节点的索引
}
}
// 堆排序
void heapSort(vector<int>& v)
{
// 建立堆
buildMaxSort(v);
// 排序
for(int i = v.size() - 1;i >= 0;-- i)
{
swap(v[0],v[i]);
maxHeapify(v,0,i - 1); // 这里不能传i,因为刚交换的那个节点我们不能去维护它的堆属性了
}
}
void printVector(const std::vector<int>& v) {
for (int num : v)
std::cout << num << " ";
std::cout << std::endl;
}
int main(int argc, char const *argv[])
{
vector<int> v = {12, 11, 13, 5, 6, 7};
heapSort(v);
printVector(v);
getchar();
return 0;
}