在面试中遇到要你手写堆排序,可以先询问是否可以使用STL中建堆的算法,问清楚面试官的意图之后方可动手写代码,交流最重要。
heap并不属于STL组件,是一个幕后英雄,STL实现了最大堆,默认的元素比较方式是less<T>,如果要使用小根堆,则将比较方式设置为greater<T>,并用来作为priority queue的底层机制。
stl中的堆默认是最大堆,要想用最小堆的话,必须要在push_heap,pop_heap,make_heap等每一个函数后面加第三个参数greater<int>(),括号不能省略。
heap是一种完全二叉树,在STL中的实现机制为:由于完全二叉树整棵树内没有任何节点漏洞,所以可以利用array来存储所有节点。通过array+一组heap算法即可以实现heap。由于heap需要动态改变大小,比如删除堆顶的元素,在堆中插入一个元素等等。所以我们用vector替代array.
STL中heap算法:
包含头文件:#include<algorithm>
push_heap算法:入堆操作
pop_heap算法:出堆操作,即将堆顶元素和二叉树最后一个元素交换,并且恢复堆。
sort_heap算法:进行堆排序
make_heap算法:使序列变成堆
从头到尾实现堆排序:
1,建堆
2,遍历数组,将堆顶元素与二叉树最后一个元素交换,然后进行堆调整操作。
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
//从i节点开始调整,n为节点总数 从0开始计算 i节点的子节点为 2*i+1, 2*i+2
void MinHeapFixdown(vector<int>& vec, int i, int n)
{
int j, temp;
temp = vec[i];
j = 2 * i + 1;
while (j < n)
{
if (j + 1 < n&&vec[j + 1] < vec[j])
j++;
if (vec[j] >= temp)
break;
vec[i] = vec[j];
i = j;
j = 2 * i + 1;
}
vec[i] = temp;
}
//建立堆
void MakeMinHeap(vector<int>& vec)
{
int size = vec.size();
//对于下标为i的节点,其父节点为(i-1)/2;所以最后一个节点的父节点下标为(size-1-1)/2,也就是size/2-1;
for (int i = size / 2 - 1; i >= 0; i--)
MinHeapFixdown(vec, i, size);
}
//堆排序
void heapSort(vector<int>& vec){
int size = vec.size();
for (int i = size - 1; i >= 1; i--)
{
swap(vec[i],vec[0]);
MinHeapFixdown(vec, 0, i);
}
}
int main()
{
int data[] = { 6, 32, 5, 3, 1, 54, 656, 44, 75 };
vector<int> vec(data,data+9);
MakeMinHeap(vec);
heapSort(vec);
for (int i = 0; i < 9; i++)
cout << vec[i]<<endl;
for (int i = 0; i < 8; i++)
cout << vec[i] << endl;
getchar();
return 0;
}