堆排序

1. 堆的概念

有n个元素的序列{k1, k2, ......, kn}，若满足下列关系：

K_i <= K_{2i &&} K_i<= K_2i+1 小顶堆  

K_i >= K_2i && K_i>= K_2i+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]的元素继续调整称为一个小顶堆          
        }

参考：严蔚敏数据结构

转载于:https://my.oschina.net/myspaceNUAA/blog/57269