/* 调整堆, 向下调整的过程 */
static void heapAdjust(std::vector<int> &vec, int start, int end)
{
int i = start;
while((2*i + 1) < end)
{
int maxIndex = 2*i + 1;
if ((2*i + 2) < end)
{
if (vec[2*i + 2] > vec[maxIndex])
{
maxIndex = 2*i + 2;
}
}
if (vec[maxIndex] > vec[i])
{
int tmp = vec[i];
vec[i] = vec[maxIndex];
vec[maxIndex] = tmp;
}
else
{
break;
}
i = maxIndex;
}
}
/* 建堆,从 vec.size() / 2 - 1开始,选择子树及自身最小的节点为当前的根节点的值,依次处理,直至根节点,全局向上调整,局部向下调整 */
static void buildHeap(std::vector<int>& vec)
{
for (int i = vec.size() / 2 - 1; i >= 0 ; i--)
{
heapAdjust(vec, i, vec.size());
}
}
/* 堆排序, 入口,时间复杂度O(nlogn),大根堆 */
static void heapSort(std::vector<int>& vec)
{
if (vec.empty())
{
return;
}
buildHeap(vec);
for (int i = vec.size() - 1; i > 0; i--)
{
int tmp = vec[i];
vec[i] = vec[0];
vec[0] = tmp;
heapAdjust(vec, 0, i);
}
}
堆排序
最新推荐文章于 2024-09-04 23:05:17 发布