上篇【堆排序(最大堆)】说过,代码是按照书中的伪码写出,书中的根结点是1,但一般我们数组的下标由0开始,故而现在改为由0开始;
上篇已说过实现的过程,故而不再赘述,细节可见上篇;
将其中的一些条件改变即可:
首先是建堆的值,由0开始的根结点,对于A[n],它的父节点是(n - 1) / 2,并且到0为止,都是父节点;
故而建堆:
for (int i = (length - 1) / 2; i >= 0; i--)
{
MaxHeapIfy(A, length, i);
}
另外在节点的两个孩子则为:
int left = i * 2 + 1; //节点i的左孩子
int right = left + 1; //节点i的右孩子节点
在进行堆排时,因为A[0]是第一个元素,故而:
for(int i = length; i >= 1;) //最后一个肯定是最小的
{
temp = A[i]; //交换堆的第一个元素和堆的最后一个元素
A[i] = A[0];
A[0] = temp;
i--; //堆的大小减一
MaxHeapIfy(A, i, 0); //调堆
}
好的,贴出代码:
#include <iostream>
#include <cstdlib>
using namespace std;
void MaxHeapIfy(int A[], int length, int i) //维护
{
int left = i * 2 + 1; //节点i的左孩子
int right = left + 1; //节点i的右孩子节点
int largest = i; //默认父节点
if (left <= length && A[largest] < A[left]) //左孩子比父节点大
{
largest = left;
}
if (right <= length && A[largest] < A[right]) //右孩子最大
{
largest = right;
}
if (i != largest) //最大值不是父节点
{
int temp = A[largest]; //exchange
A[largest] = A[i];
A[i] = temp;
MaxHeapIfy(A, length, largest); //继续维护
}
}
void BuildMaxHeap(int A[], int length) //建堆
{
for (int i = (length - 1) / 2; i >= 0; i--)
{
MaxHeapIfy(A, length, i);
}
}
void HeapSort(int A[], int length) //堆排
{
int temp;
BuildMaxHeap(A, length); //建堆
cout<<"建堆情况:"; //
for(int i = 0; i <= length; i++)
cout<<A[i]<<" ";
cout<<endl;
for(int i = length; i >= 1;) //最后一个肯定是最小的
{
temp = A[i]; //交换堆的第一个元素和堆的最后一个元素
A[i] = A[0];
A[0] = temp;
i--; //堆的大小减一
MaxHeapIfy(A, i, 0); //调堆
}
}
int main()
{
int A[] = {0, 4, 1, 3, 2, 16, 9, 10, 14, 8, 7};
/*int* A = new int[1000];
A[0] = 0;
for(int i = 1; i < 1000; i++)
A[i] = rand()%10000 + 1;*/
int length = sizeof(A) / sizeof(int); //
//int length = 1000;
HeapSort(A, length - 1);
cout<<"排序结果:";
for(int i = 0; i < length; i++) //cout
cout<<A[i]<<" ";
cout<<endl;
return 0;
}
欢迎交流,o(∩_∩)o