堆的存储:一般用数组表示堆
i 结点的父结点(i-1)/2,左右结点下标分别为2*i+1和2*i+2
例如:
基本思想:
- 将待排序列构造成一个大顶堆或者小顶堆。
- 最大值就是对顶的根节点,将它移走将其数堆中的元素进行交换,此时末尾元素就是最大值。
- 然后将剩余的n-1个序列重新排序构成一个堆,就会得到n个元素中的最大值。
- 反复进行。
注意:
- 建堆的过程是从最后一个子树开始(高楼平地起)
- 每次调整的时候都是从要调整的数的根位置开始
建堆:从第一个非叶子节点开始,将三个数最大的数放入根的位置。比如 0 1 2,2是最大的数,将2调整到根的位置。
调整过程:每次调整都是从要调整的树的根位置开始。每次都将最大的值的放在根的位置。
代码:
//堆排序
void Heap_adjust(int arr[], int p,int m)
{
int temp = arr[p];//将根节点值保存,以便后面和最大值要交换
int j;
for (j = p * 2 + 1; j <= m; j = j * 2 + 1)//j为左孩子,循环一次之后,j向下层移动
{
if (j + 1 <= m&&arr[j + 1] > arr[j])//比较左右孩子大小,j下标的值最大
{
j++;
}
if (temp > arr[j])//若根节点值比最大的孩子节点值大则退出,不用交换
{
break;
}
arr[p] = arr[j];//存在孩子节点比根节点值大,交换两个值
p = j;//将下标赋给根
}
arr[p] = temp;//将temp中的值放在合适位置
}
void Make_heap(int arr[], int len)//建堆
{
for (int i = len / 2 - 1; i >= 0; i--)//从倒数第一个非叶子节点开始
{
Heap_adjust(arr, i, len);//调整
}
}
void swap(int arr[], int a, int b)
{
int temp = arr[a];
arr[a] = arr[b];
arr[b] = temp;
}
void Heap_sort(int arr[], int len)
{
Make_heap(arr,len);//建堆
for (int i = len - 1; i >= 0; i--)从最后一个叶子开始交换数据
{
swap(arr, 0, i);//和根节点的最大值交换
Heap_adjust(arr, 0, i - 1);//调整
}
}
void main()
{
int arr[] = { 5, 9, 8, 7, 10, 2, 1 };
int len = sizeof(arr) / sizeof(arr[0]);
Heap_sort(arr, len);//调用堆排序
for (int i = 0; i<len; ++i)
{
cout << arr[i] << " ";
}
cout << endl;
}