1. 堆的概念
Ki <= K2i && Ki<= K2i+1 小顶堆
Ki >= K2i && Ki>= K2i+1 大顶堆
注: (当然前提是在左右孩子存在的情况下,其实这种要求是针对非叶子节点)
2. 堆排序方法:
此处采用小顶堆:
由于堆是逻辑二叉树(是指概念上采用二叉树形式组织,但是在内存中采用线性表的方式存储)
每次将堆调整为一个小顶堆,则堆顶为最小元素,找到最小元素后对剩下的元素进行堆调整,重新得到一个最小堆,逐渐找到次小的元素,将一个无序的数组调整为有序的数组。
那么利用堆的概念进行排序的关键是:
1.建堆(现在只考虑最小堆)! 如何将一个无序数组建成小顶堆,就能找最小元素(其实概念与选择排序相似,每次都找最小元素)
2. 在输出堆顶后,如何对剩下的元素进行调整,重新成为一个最小堆,就能找到次小元素
先解决第二个问题:
对于数组A[1......n],已经是最小堆,堆顶为最小值,现在输出最小值,将A[1]和A[n]交换,这时A[1......n-1]的最小堆特征被破坏,需要调整。从A[1]开始进行调整,
若孩子节点中的最小者小于父节点的value,则与父节点交换,父节点下沉。若父节点下沉后以父节点为根的子树依旧不符合最小堆的特征,继续调整。该流程可以将整个堆调整为一个最小堆。
解决第一个问题:
从第一个非叶子节点开始调整(叶子节点就一个,可以看成一个合理的堆),这时只有堆顶不符合小顶堆的要求,按照上述方法进行调整。调整完后往前推,继续调整(每次都是堆顶不符合最小堆的要求,其他元素都符合)。继续调整,一直到数组的第一个元素,调整完毕,整个数组被建成一个最小堆。
图形演示调整堆的过程:
根据堆的概念,先上个图,看看是如何将一个无序的数组调整称为一个小顶堆。
就不展示整个排序流程了。只要将上图的13(堆顶)和97(最后一位)交换,在调整A[1.....n-1]即可。
代码:
/*******************************************************************************
函数名: HeadAdjust
功能: 将数组pSrcData[p......r]调整为小顶堆
注: 数组中除了pSrcData[p],其他元素都符合小顶堆的要求
返回值:无
********************************************************************************/
void HeadAdjust(int* pSrcData, int p, int r)
{
int nCurIndex = p; //代表当前元素下标
int nHeadNodeVal = pSrcData[p]; //记录堆顶元素
//节点p的孩子节点下标为2*p, 2*p+1
//每循环一次,若childNode.val < nCurVal 则当前元素为孩子节点中较小的孩子
for(int i = 2*p; i <=r; i = i * 2)
{
int nLeftIndex = i;
int nLeftVal = pSrcData[nLeftIndex];
int nRightIndex = nLeftIndex + 1;
if (nRightIndex <= r) //右孩子存在
{
int nRightVal = pSrcData[nRightIndex];
if (nLeftVal > nRightVal) //i指向较小的孩子节点
i++;
}
if (nHeadNodeVal<= pSrcData[i]) //当前节点比孩子节点都要小,无需调整
break;
pSrcData[nCurIndex] = pSrcData[i]; //小元素上移
nCurIndex = i;
}
if (nCurIndex != p)
pSrcData[nCurIndex] = nHeadNodeVal;
}
void HeapSort(int* pSrcData, int nLen)
{
//从第一个非叶子节点开始进行建堆,得到一个小顶堆
for (int i = nLen/2; i >0; i--)
HeadAdjust(pSrcData, i, nLen-1);
for (int i = nLen-1; i >1; i--)
{
//将最小一个元素(第一个元素)和最后一个元素互换
int nTmp = pSrcData[i];
pSrcData[i] = pSrcData[1];
pSrcData[1] = nTmp;
HeadAdjust(pSrcData, 1, i-1); //对[1....i-1]的元素继续调整称为一个小顶堆
}
}
结果演示:
//从第一个非叶子节点开始进行建堆,得到一个小顶堆
for (int i = nLen/2; i >0; i--)
HeadAdjust(pSrcData, i, nLen-1);
for (int i = nLen-1; i >1; i--)
{
//将最小一个元素(第一个元素)和最后一个元素互换
int nTmp = pSrcData[i];
pSrcData[i] = pSrcData[1];
pSrcData[1] = nTmp;
HeadAdjust(pSrcData, 1, i-1); //对[1....i-1]的元素继续调整称为一个小顶堆
}