5. 堆排序

思想

维护一种数据结构,它能够完成下面两个功能:插入数据;取出最小数据。java

由于数组和链表均不能均衡这两个功能的时间复杂度。因而采用彻底二叉树来实现这两种功能。数组

小堆,每一个节点值都小于其子节点的值。数据结构

其实,就有点像排行榜的感受,当有新的元素登上排行榜的时候,将会去除最后一名的元素,新加入的元素存储在相应的位置。code

只不过这个排行榜是用二叉树堆实现的,这个二叉树堆也不想二叉搜索树那样彻底有序。其只要求父节点小于子节点就能够。排序

由于二叉树堆的特性,能够利用数组结构进行实现。队列

利用堆排序能够完成Top k问题,就像排行榜同样。it

由于二叉树堆,更像是数据结构,因此这里不去实现其具体的结构,而是使用Java中实现的优先队列,PriorityQueue。io

实现

import java.util.Arrays;
import java.util.PriorityQueue;

public class HeapSort {
    public static void main(String[] args) {
        int[] nums = {5, 7, 2, 4, 1, 6, 8, 9};
        heapSort(nums);
        System.out.println(Arrays.toString(nums));
    }

    public static void heapSort(int[] nums){
        PriorityQueue<Integer> heap = new PriorityQueue<>();
        for(int i = 0; i < nums.length; i++){
            heap.offer(nums[i]);
        }

        for (int i = 0; i < nums.length; i++) {
            nums[i] = heap.poll();
        }
    }
}

复杂度

堆排序的时间复杂度,在于:插入和去除元素后,对堆的维护,时间复杂度为NlogN。class

空间复杂度为O(n)import

相关文章
相关标签/搜索