一、建堆。
实现思路:
1、首先将需要实现排序的数组建堆。即将数据放在堆中。
2、然后对堆进行调整,使其成为大根堆或小根堆(本次实现大根堆):
取k/2位置的节点为初始值,向上便利每一个节点,并判断该节点是否大于子节点中的最大值(左子节点的索引为2k),右子节点的索引为(2k+1);如果大于,则交换该节点和最大子节点的位置。并继续向下判断交换后的节点的位置是否正确(是否满足大于最大子节点);
3、遍历完后,i–继续向上便利,直到根节点。
二、堆排序
实现思路:
1、首先将堆的根节点和最后一个子节点交换,取出最大值,即根节点的值。交换完后由于打乱的堆的顺序,需要重新调正堆,所以从根节点,依次向下遍历,知道调整完,但是交换后的最大节点不参与调整。
class Heap{
private Integer[] arr;
private Integer len = 0;
public Heap(Integer[] arr) {
this.len = arr.length;
this.arr = new Integer[len+1];
System.arraycopy(arr, 0, this.arr, 1, len);
}
public void bulidHeap() {
//建堆
for(int i=len/2;i>=1;i--) {
this.adjustHeap(i,len);
}
}
//堆排序
public void sortHeap() {
for(int i =1;i<arr.length;i++) {
//交换第一个元素(即最大值)和最后一个元素(不包括已排序的)
int temp = arr[1];
arr[1] = arr[arr.length-i];
arr[arr.length-i] = temp;
adjustHeap(1,arr.length-i-1);
}
}
//调整堆,k是初始的节点,range是调整的范围。
public void adjustHeap(int k,int range) {
while(2*k<=range) {
int max =0;
//改节点没有右节点
if(2*k+1>range) {
max = 2*k;
}else {
if(arr[2*k]>arr[2*k+1]) {
max = 2*k;
}else {
max = 2*k+1;
}
}
if(arr[k]<arr[max]) {
Integer temp = arr[k];
arr[k] = arr[max];
arr[max] = temp;
}
k = max;
}
}
}