堆 (Heap)

“堆”这种数据结构常用在“优先级队列”的实现上, 比如Java中的PriorityQueue。今天讲讲什么是堆,如何去实现一个堆。


</br>
</br>

什么是堆

  • 堆是一个二叉树。
  • 堆的任意节点,总大于(小于)子节点。(大于即为大根堆,小于即为小根堆,根节点是最值)
  • 而且堆总是完全树。(除了最底层,其它层的节点都被塞满)

下图即为一个大根堆:


堆的数组表现形式.jpg

</br>
</br>

数组实现堆

在程序中,堆通常使用数组来实现(因为增删频繁)。并遵循下面的规则:

  • 元素的父元素:(index-1)/2
  • 元素的左孩子:(index*2)+1
  • 元素的右孩子:(index*2)+2

</br>
</br>

堆的操作

对于堆,通常有下面几种常见操作:

  • 构建堆 (给定一个数组,按照堆的逻辑去重构它)
  • 删除根节点 (从堆中取出最值)
  • 插入一个节点 (加入一个节点到堆中,当然要重构堆)

三种操作,但在实现的时候主要涉及到两个核心方法,adjustDown向下调整,和adjustUp向上调整。


</br>
</br>

adjustDown

    //向下调整
    public void adjustDown(int current) {
        int right = getRightChildIndex(current);// 右孩子
        int left = getLeftChildIndex(current);// 左孩子
        int max = findMax(current, right, left);// 三个节点,值最大的index
        if (max != current) {
            // 需要调整
            swap(current, max);
            // 递归调整与根节点进行交换的节点,保证下层也是堆
            adjustDown(max);
        }
    }
    // 三者最大的节点的索引
    private int findMax(int node, int right, int left) {
        int max = node;
        if (right < length && data[right] > data[max]) {
            max = right;
        }
        if (left < length && data[left] > data[max]) {
            max = left;
        }
        return max;
    }
    //交换数据
    public void swap(int n1, int n2) {
        int temp = data[n1];
        data[n1] = data[n2];
        data[n2] = temp;
    }

</br>
</br>

构建堆

构建堆时用到了adjustDown,从底部非叶子节点开始,每到某一节点,都会向下去调整,使得这个节点和其子节点形成堆。

    protected int[] data;
    protected int length = 0;

    public MaxHeap(int[] data) {
        this.data = data;
        this.length = data.length;
        // 从最后一个节点的父节点开始构建堆
        int start = getParentIndex(length - 1);
        for (; start >= 0; start--) {
            adjustDown(start);
        }
    }
传入一个数组
未调整过的堆
index为3的地方,是最后一个非叶子节点,从这个节点开始构建堆
为index为2的地方构建堆
为index为1的地方构建堆
为index为1的地方,继续向下递归构建堆,即对index为3的地方构建堆

可以看到,在对index为1的地方构建堆时,因4和18的位置调换了,子堆可能已不符合要求,所以必须递归调用去重新构建子堆。这就是关键。


</br>
</br>

删除根节点

    public MaxHeap remove() {
        // 将最后一个节点与头结点交换
        swap(0, length - 1);
        // 重新复制一个数组
        int[] newData = new int[length - 1];
        System.arraycopy(data, 0, newData, 0, length - 1);
        this.data = newData;
        this.length = length - 1;
        // 只需要从根节点,向下调整一次即可
        adjustDown(0);
        return this;
    }

</br>
</br>

增加一个节点

    public MaxHeap insert(int value) {
        // 插入到数组最后
        int[] newData = new int[length + 1];
        System.arraycopy(data, 0, newData, 0, length);
        newData[length] = value;
        this.data = newData;
        this.length = length + 1;
        // 从最后一个节点开始自下而上调整堆(只要比父节点大,就交互)
        adjustUpHeap(this.length - 1);
        return this;
    }

    public void adjustUpHeap(int node) {
        int parent = getParentIndex(node);// 父节点
        if (parent >= 0 && data[parent] < data[node]) {
            swap(node, parent);
            adjustUpHeap(parent);// 递归调整与根节点进行交换的节点,保证上层也是堆
        }
    }

</br>
</br>

时间复杂度

堆的各种操作

</br>
</br>

代码

Code


</br>
</br>

参考

数据结构系列——堆

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,907评论 6 506
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,987评论 3 395
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 164,298评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,586评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,633评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,488评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,275评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,176评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,619评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,819评论 3 336
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,932评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,655评论 5 346
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,265评论 3 329
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,871评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,994评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,095评论 3 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,884评论 2 354

推荐阅读更多精彩内容