public class BigHeap2 {
public static void main(String[] args) {
BigHeap2 bigHeap2 = new BigHeap2();
int[] a = new int[]{5, 4, 3, 2, 1, 10, 20, 41, 15};
bigHeap2.sort(a);
System.out.println(Arrays.toString(a));
}
/**
* [1,3,5,2,9,6,15]
* arr[i] < arr[2 * i + 1] && arr[i] < arr[2 * i + 2]
*/
public void sort(int[] arr) {
for (int i = arr.length / 2 - 1; i >= 0; i--) {
adjustHeap(arr, i);
}
}
public void adjustHeap(int[] arr, int index) {
int temp = arr[index];
//左节点指针
int cIndex = 2 * index + 1;
//如果当前index的左子节点不存在就返回。arr判断是不是大于最大索引
if(cIndex > arr.length -1)
return;
//判断右子节点存在,如果右节点存在并且大于左节点,指针指向右节点
if (cIndex + 1 < arr.length && arr[cIndex + 1] > arr[cIndex]) {
cIndex++;
}
if (arr[index] < arr[cIndex]) {
//交换节点
arr[index] = arr[cIndex];
arr[cIndex] = temp;
}
//交换节点后,需要判断交换后的节点是否符合要求
adjustHeap(arr, cIndex);
}
}
大堆顶实现
最新推荐文章于 2024-06-03 22:34:19 发布