趁着昨天写了一个前k个最高频元素,用到了堆排序O(nlogn),复习一下。
class Solution {
public void heapSort(int[] arr){
buildMaxHeap(arr);
//把最大值放到末尾,len--,再将最大值放到末尾,len--,形成一个升序的arr数组,完成heapsort
for(int i = arr.length - 1;i > 0;i--) {
swap(arr,0,i);
len--;
heapify(arr,0);
}
return arr;
}
//把所有父节点遍历即产生最大堆
public void buildMaxHeap(int[] arr){
len = arr.length;
for(int i = len / 2 - 1;i > = 0;i++) {
heapify(arr,i);
}
}
//调整堆结构,把最大值赋值到父节点,如果子节点比父节点大,则交换后继续对交换后的子节点进行调整(递归调用)。
public void heapify(int[] arr, int i){
int left = 2 * i + 1;
int right = 2 * i + 2;
//用来存储最大值
int largest = i;
int len = arr.length;
if(left < len && arr[left] > arr[largest]){
largest = left;
}
if(right < len && arr[right] > arr[largest]){
largest = right;
}
if(largest != i) {
swap(arr,i,largest);
heapify(arr,largest);
}
}
public void swap(int[] arr, int i, int j){
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
}
至于为什么第一个非叶子节点是(len - 1) / 2:
在完全二叉树中, 第一个非叶子结点 其实就是 最后一个叶子结点的父节点。
假定父节点为i;则 其左叶子为2i+1 , 右叶子为2i+2;
则当叶子节点为n-1时,当n为偶数,即n是左孩子时,父节点 i = (n-1-1)/2 = n/2 - 1.
当n为奇数,即n是右孩子时,父节点 i = (n-1-2)/2 = n/2 - 1.因为n是奇数,且是向下取整。