二叉树可以用数组形式表示:
- 左孩子:2*i+1
- 右孩子:2*i+2
- 一个孩子的父节点:i-1/2
堆-----》完全二叉树,堆可以分为大根堆和小跟堆
大根堆:任何一个节点包括它的孩子节点,都是父节点最大
HeapInsert:
给定一个数组,如何改成大根堆:(a[0]已经是大根堆了,再加入a[1],只需要不断往上比较a[1]的父节点i-1/2,若当前节点比父节点大,则交换,直到小于或等于父节点)
public class HeapSore01 {
public static void heapSort(int[] arr) {
if (arr == null || arr.length < 2) {
return;
}
for (int i = 0; i < arr.length; i++) {
heapInsert(arr, i);//比较0~i
}
}
public static void heapInsert(int[] arr, int index) {
while (arr[index] > arr[(index - 1) / 2]) {//直到当前节点不比父节点大,当当前节点为跟节点时也成立,a[0]不大于a[0]
swap(arr, index, (index - 1) / 2);
index = (index - 1) / 2;
}
}
public static void swap(int []arr,int l,int r){
int tmp=arr[l];
arr[l]=arr[r];
arr[r]=tmp;
}
public static void main(String[] args) {
int[] arr={2,1,3,6,0,4};
System.out.println(Arrays.toString(arr));
heapSort(arr);
System.out.println(Arrays.toString(arr));
}
}
建立一个大根堆的时间复杂度O(N),由于每个节点加进来,只需要往上查找父节点进行比较,所以与层数有关,堆又是完全二叉树,则n的层数的logn
所以大根堆的时间复杂度为:log1+log2+log3+...+log(i-1)=O(N),因为第i个最多只需要遍历log(i-1)层
heapify(假设一个大根堆中有一个数变小,那么如何重新调整为大根堆)
找到变小的节点的孩子,跟最大的孩子交换,不断找交换后的孩子,直
到没有孩子或者比孩子比父节点小结束
代码实现:
public static void heapSort(int[] arr) {
if (arr == null || arr.length < 2) {
return;
}
for (int i = 0; i < arr.length; i++) {
heapInsert(arr, i);//先将一个数组变成一个大根堆
}
int size = arr.length;
swap(arr, 0, --size);
while (size > 0) {
heapify(arr, 0, size);
swap(arr, 0, --size);
}
}
public static void heapInsert(int[] arr, int index) {
while (arr[index] > arr[(index - 1) / 2]) {
swap(arr, index, (index - 1) / 2);
index = (index - 1) / 2;
}
}
//若index位置的数变小,那么如何重新排序成大根堆
public static void heapify(int[] arr, int index, int size) {
int left = index * 2 + 1;//先找左孩子
while (left < size) {//左孩子没有越界
int largest = left + 1 < size && arr[left + 1] > arr[left] ? left + 1 : left;//若右孩子没有越界且大于左孩子,那么就取右孩子否则取左孩子
largest = arr[largest] > arr[index] ? largest : index;//比较父节点和子节点的最大值,并交换最大值
if (largest == index) {//若没有发生交换,这说明已经排好序,跳出
break;
}
swap(arr, largest, index);//若没有排好序,则交换,然后下面再修改值,进行下次比较
index = largest;
left = index * 2 + 1;
}
}
public static void swap(int[] arr, int i, int j) {
int tmp = arr[i];
arr[i] = arr[j];
arr[j] = tmp;
}
减堆操作:若想把堆顶元素弹出,直接把堆顶元素和最后一个元素交换,然后长度-1然后堆顶元素经历一次heapify过程,即可把堆调整有序
堆排序:
- 先把给定的数变成大根堆
- 将顶部和最后一个交换,数组长度标识-1,这样就得到了当前堆最大的值
- 若堆中超过1个数,则重新排序成大根堆,即heapify过程,然后再重复2过程
- 经过上述操作后,最终得到一个从小到大排序的数组