数据结构与算法--堆

214 阅读5分钟

基础概念

堆的定义

堆是计算机科学中一类特殊的数据结构的统称,堆通常可以被看做是一棵完全二叉树的数组对象。

堆的特性

  1. 它是完全二叉树,际了树的最后一层结点不需要是满的,其它的每一层从左到右都是满的,如果最后一层结点不是满的,那么要求左满右不满

image.png

  1. 用数组实现(且通常从索引为1处开始插入,废弃了索引0)
  • 将二叉树的结点按照层级顺序放入数组中,根结点在位置1,它的子结点在位置2和3,而子结点的子结点则分别在位置4,5,6和7,以此类推。
  • 如果一个结点的位置为k,则它的父结点的位置为k/2,而它的两个子结点的位置则分别为2k2k+1。这样,在不使用指针的情况下,我们也可以通过计算数组的索引在树中上下移动:从a[k]向上一层,就令k等于k/2,向下一层就令k等于2k或2k+1。

image.png

  1. 每个结点都大于等于它的两个子结点。这里要注意堆中仅仅规定了每个结点大于等于它的两个子结点,但这两个子结点的顺序并没有做规定,跟我们之前学习的二叉查找树是有区别的。

堆的种类

  1. 大顶堆:arr[i] >= arr[2i] && arr[i] >= arr[2i+1]
  • 每个结点的值都小于等于它的子节点
  1. 小顶堆:arr[i] <= arr[2i] && arr[i] <= arr[2i+1]
  • 每个结点的值都大于等于它的子节点

堆的构造(这里以构造大顶堆为例来构造)

堆的API设计

image.png

重要方法方法思路详解

  1. insert -- 往堆中插入元素
  • 直接往数组中放入值,破坏了原本的堆的有序结构,使用上浮算法调整
  1. delMax -- 删除最大元素
  • 因为删除最大元素,会改变堆的有序结构,所以我们要先交换要删除最大的元素和最后一个元素的位置
  • 使用下沉算法,调整堆,使堆成为有序结构
  • 然后此时最后一个元素就是我们要删除的元素,我们直接让它指向null,这样就实现了删除最大的元素
  • 元素个数减一
  1. swim -- 上浮算法
  • 不断让插入的值和和它的父节点比,比父节点大,就上移,直到比父节点小,比父节点小的时候,插入值的位置就是正确的了
  1. sink -- 下沉算法
  • 循环遍历从k开始,找出k的两个子节点中较大的那个
  • 如果子节点中较大的那个比k大,就交换他们的位置,否则k的位置就是正确的

代码实现

package com.study.Heap;

public class heapAPI<T extends Comparable<T>> {
    // 用数组存储堆中的数据
    private T[] items;
    // 记录堆中的元素个数
    private int N;

    // 初始化数组的大小
    public heapAPI(int capacity) {
        // capacity+1 是因为,我们废弃掉了数组中索引为0的位置
        this.items = (T[]) new Comparable[capacity + 1];
    }

    // 堆的元素个数
    public int size() {
        return N;
    }

    // 判断堆中索引i处的元素是否小于索引j处的元素
    private boolean less(int i, int j) {
        return items[i].compareTo(items[j]) < 0;
    }

    // 交换堆中索引i和索引j处的元素
    private void exchange(int i, int j) {
        T temp;
        temp = items[i];
        items[i] = items[j];
        items[j] = temp;
    }

    // 在堆中插入一个元素
    public void insert(T t) {
        // ++N 第一个元素存放在数组的1位置,0位置的索引被我们废弃了 至于为什么,我也不知道
        items[++N] = t;
        // 插入完,堆结构被破坏,需要使用上浮算法调整堆结构
        swim(N);
    }

    // 使用上浮算法,使得插入的元素放在堆中的一个正确的位置
    /*
        上浮算法:让插入的元素和它的父节点比,比父节点大,就上移,直到比父节点小
     */
    private void swim(int k) {
        while (k > 1) {
            // 插入的元素的值比它的父节点大,交换。
            if (less(k / 2, k)) {
                exchange(k / 2, k);
            } else {
                break;
            }
            k = k / 2;
        }
    }

    // 删除堆中最大的元素,并返回这个最大元素
    // 先交换要删除最大的元素和最后一个元素的位置
    // 然后此时最后一个元素就是我们要删除的元素,我们直接让它指向null
    // 元素减一
    public T deleteMax() {
        T t = items[1];
        exchange(1, N);
        sink(1);
        items[N--] = null;
        return t;
    }

    // 使用下沉算法,调整删除堆中的元素后,堆的结构
    /*
        下沉算法:
               1. 循环遍历从k开始,找出k的两个子节点中较大的那个
               2. 如果子节点中较大的那个比k大,就交换他们的位置,否则k的位置就是正确的
     */
    private void sink(int k) {
        while (2 * k <= N) {
            int max;
            // 找出子节点中较大的结点所在的索引
            if (2 * k + 1 <= N) {
                if (less(2 * k, 2 * k + 1)) {
                    max = 2 * k + 1;
                } else {
                    max = 2 * k;
                }
            } else {
                max = 2 * k;
            }
            // 当前结点和子结点比较
            if (less(k, max)) {
                exchange(k, max);
                k = max;
            } else {
                break;
            }

        }
    }

}

测试

package com.study.Heap;

public class heapTest {
    public static void main(String[] args) {
        heapAPI<String> heap = new heapAPI<>(20);
        heap.insert("T");
        heap.insert("S");
        heap.insert("R");
        heap.insert("P");
        heap.insert("N");
        heap.insert("O");
        heap.insert("A");
        heap.insert("E");
        heap.insert("I");
        heap.insert("H");
        heap.insert("G");
        System.out.println(heap.size());
        String s = heap.deleteMax();
        System.out.println(s);
        System.out.println(heap.size());
    }
}

String s = heap.deleteMax();处打断点,debug一下,看一下删除后,堆有没有调整好

删除前的堆

image.png

删除最大元素后的堆

image.png

完美实现

堆排序

思想

将待排序序列构造成一个大顶堆,此时,整个序列的最大值就是堆顶的根节点。将其与末尾元素进行交换,此时末尾就为最大值。然后将剩余n-1个元素重新构造成一个堆,这样会得到n个元素的次小值。如此反复执行,便能得到一个有序序列了

步骤

  1. 构造初始堆。将给定无序序列构造成一个大顶堆(一般升序采用大顶堆,降序采用小顶堆)。
  2. 将堆顶元素与末尾元素进行交换,使末尾元素最大。然后继续调整堆,再将堆顶元素与末尾元素交换,得到第二大元素。如此反复进行交换、重建、交换。
  3. 再给一个详解堆排序的链接

代码实现(这里没有废弃数组的0索引)

package com.study.sort.AdvanceSort;

// 堆排序
public class Heap {
    public static void main(String[] args) {
        int[] a = {2, 5, 2, 3, 5, 3, 10, 7, 9};

        //构建大顶堆
        for (int i = a.length / 2 - 1; i >= 0; i--) {
            adjustHeap(a, i, a.length);
        }

        // 调整大顶堆,交换堆顶元素和末尾元素
        for (int i = a.length - 1; i > 0; i--) {
            swap(a, 0, i);//交换堆顶元素和末尾元素
            adjustHeap(a, 0, i);//重新调整堆
        }

        // 打印排序结果
        System.out.println("使用堆排序的结果:");
        for (int i = 0; i < a.length - 1; i++) {
            System.out.print(a[i] + ",");
        }
        System.out.print(a[a.length - 1]);
    }

    private static void adjustHeap(int[] a, int i, int length) {
        int temp = a[i];
        for (int j = 2 * i + 1; j < length; j = 2 * j + 1) {
            if (j + 1 < length && a[j] < a[j + 1]) {
                j++;
            }
            if (a[j] > temp) {
                a[i] = a[j];
                i = j;
            } else {
                break;
            }
        }
        a[i] = temp;
    }

    private static void swap(int[] a, int m, int n) {
        int temp = a[m];
        a[m] = a[n];
        a[n] = temp;
    }

}

结果:

image.png