堆排序模板
class HeapSort {
// 堆排序
public void heapSort(int[] tree, int n) {
buildHeap(tree, n);
for (int i = n - 1; i > -1; i--) {
swap(tree, 0, i);// 把最大的放在最后面
heapify(tree, 0, i);
}
}
// 构建最大堆
public void buildHeap(int[] tree, int n) {
int lastNode = tree.length - 1;
int parent = (lastNode - 1) / 2;//最后一个节点的父节点
// 从下往上构建
for (int i = parent; i >= 0; i--) {
heapify(tree, i, n);
}
}
// 找出最大值交换
public void heapify(int[] tree, int i, int n) {
int max = i;
int c1 = 2 * i + 1;
int c2 = 2 * i + 2;
if (c1 < n && tree[c1] > tree[max])max = c1;
if (c2 < n && tree[c2] > tree[max])max = c2;
if (max != i) {
swap(tree, i, max);
heapify(tree, max, n);
}
}
// 交换
public void swap(int[] arr, int i, int j) {
int tmp = arr[i];
arr[i] = arr[j];
arr[j] = tmp;
}
}
思考
堆和排序是两个概念
我们是先构建出大顶堆或者小顶堆之后
再在堆的基础上排序,最终返回正确的顺序
首先构造出一个大顶堆,这样最上面就是最小值
之后将最小值和(变换后)数组最后一个值交换,
然后将剩余n-1个元素重新构造成一个堆
这样 经过n次交换,就构成了一个从小大大的序列