数据结构系列x-堆

Heap

参照

一个优先级队列。PriorityQueue便是根据堆来实现的

找出前K个数(从大到小),构建一个容量为K的小堆,遍历序列,如果元素比堆顶的元素大,则替换之,然后下沉,维护堆。

还有一个应用场景,堆排序。这都是以前

堆的数据存储、增删改查遍历操作实现

堆是一个完全二叉树,何为完全二叉树,1.所有的叶节点都出现在最下边两层,不就是平衡二叉树吗。任一两条路径的高度差不超过1。

最大堆:节点的值大于子节点的值(子节点之间不一定满足何种顺序),因此根节点是最大的

最小堆:节点的值小于子节点的值,因此根节点是最小的

堆整体趋势上是有序的,但不严格有序,上一层所有节点值大于下一层所有节点的值,但是一层内节点间是无序的

堆的数据存储也可以用节点,和树一样通过维护节点关系来维护,但是因为堆是一个完全二叉树的前提,父子节点间的节点数量满足一定的关系,所以可以用数组来存储。找父节点、左节点和右节点可以通过计算数组下标来获取。这是可以用数组来存储的必要条件。如果用节点关系的话,每个节点需要维护左右节点和父节点的引用,增加存储空间,二是同一排的节点之间没有直接的引用,不便于遍历。所以用数组存储各个节点。

通过一个节点获取左、右、父节点的下标计算公式推导。如果获取前一个或后一个直接将下标减1或加1即可。

定位一个父节点,比如在堆中的第c排,第l个。推导其左节点下标和自己的下标关系。

堆.jpg

上图即为节点在堆中的位置和映射到数组的位置关系。

一个标准的堆,数字代表节点在堆中坐标,横坐标记为c,纵坐标记为l,前提准备,需要用到等比数列表达式和求和公式,这里不展开。

对于第c层,这一层的节点数量为2^{c-1},

1到c层总节点数量为2^c-1,推导i_左i_右i_父i的关系

N前面的节点数量

2^{c-1}-1+l-1=2^{c-1}+l-2,则i=2^{c-1}+l-2(下标从0开始)

N到N左之间的节点数量

2^{c-1}-l+2*(l-1)=2^{c-1}+l-2

可以看出N和N左之间的节点数量和N之前的节点数量是一样的,记为n

i=n

i_左=n+1+n=2*n+1

i_左=2*i+1

i_右=i_左+1=2*i+2

通过父节点找两个子节点的关系

i_左=2^{c-1}+l-2(下标从0开始)

通过上面例子,假定父节点为N_{3,3},则c=3,l=c

计算得

i=2^{3-1}+3-2=5

i_左=2^*5+1=11

跟示意图吻合

通过子节点找父节点,因为子节点可能是左节点,也有可能是右节点。最终我们将要推导出一个公式,两个子节点的坐标参数都能计算出唯一的父节点坐标,先根据左节点和右节点分别求出父节点的下标计算公式

通过左节点

i_父=(i_左-1)/2

通过右节点

i_父=(i_右-2)/2

备注:此处的计算结果按照程序中的除法概念,只取整数。因为堆中没有直接维护节点间的关系,对于任意子节点,无从得知其是左节点还是右节点。所以这个两个公式哪个可以作为两种子节点下标计算父节点下标的通用公式?

按照堆是一个完全二叉树设定,从第二层开始,每一层的节点数量都是上一层的2倍,即每一层的节点数都是偶数。所以任意一个右节点都处于第奇数个位置上,即在数组的下标一定是偶数,同理得左节点在数组中的下标一定是奇数。即i_左\%2=1i_右\%2=0。假设用左节点的公式来计算右节点下标,即

(i_右-1)/2 =(i_左-1+1)/2

​ 因为i_左\%2=1,所以i_左/2=(i_左-1)/2,多加的1作为余数被丢弃掉了

所以

(i_右-1)/2=(i_左-1)/2

即可以用左节点公式来计算右节点下标。

反之,用右节点公式计算左节点,因为(i_右-2)/2\%2=0(刚好除尽),但是(i_左-2)/2=(i_右-1-2)/2,不能被整除,且整数减少了1。所以右节点计算公式不能计算左节点。

综上,通过任意子节点的下标计算父节点的下标能且只能通过左节点公式来计算。

i_父=(i_子-1)/2

以上父子节点下标互相计算公式,通过JDK PriorityQueue源码也能得到印证

private void siftUpComparable(int k, E x) {
    Comparable<? super E> key = (Comparable<? super E>) x;
    while (k > 0) {
        int parent = (k - 1) >>> 1;// 通过子节点下标计算父节点下标
        Object e = queue[parent];
        if (key.compareTo((E) e) >= 0)
            break;
        queue[k] = e;
        k = parent;
    }
    queue[k] = key;
}
private void siftDownComparable(int k, E x) {
        Comparable<? super E> key = (Comparable<? super E>)x;
        int half = size >>> 1;        // loop while a non-leaf
        while (k < half) {
            int child = (k << 1) + 1; // assume left child is least 通过父节点下标计算左节点下标
            Object c = queue[child];
            int right = child + 1;
            if (right < size &&
                ((Comparable<? super E>) c).compareTo((E) queue[right]) > 0)
                c = queue[child = right];
            if (key.compareTo((E) c) <= 0)
                break;
            queue[k] = c;
            k = child;
        }
        queue[k] = key;
    }

以小顶堆为例

新增操作

先将值追加到堆尾,即放入到数组的size下标处,然后递归跟父节点比较,如果值比父节点小,则向上筛选,即跟父节点互换,代码实现如下

private Integer[] queue;

    //节点个数
public void add(int value) {
    //放入堆底
    int k = size;
    queue[k] = value;
    size++;
    if (k == 0) {
        return;
    }
    //向上翻
    siftUp(k, value);
}

private void siftUp(int k, int value) {
        while (k > 0) {
            int parent = (k - 1) >>> 1;//为什么是无符号右移
            if (queue[parent] == null) {
                throw new IllegalArgumentException("parent is null when add " + value);
            }
            if (value < queue[parent]) {
                //对换
                queue[k] = queue[parent];
                queue[parent] = value;
                //继续往上找
                k = parent;
            }else {
                break;
            }

        }
    }

删除操作

先把最后一个元素放置在待删除的位置,然后下调,直至子节点比当前节点都大或者没有子节点了,代码实现如下

public boolean remove(int v) {
    int p = indexOf(v);
    if (p == -1) {
        return false;
    }
    removeAt(p);
    return true;
}

private int indexOf(Integer o) {
    if (o != null) {
        for (int i = 0; i < size; i++)
            if (o.equals(queue[i]))
                return i;
    }
    return -1;
}
public int removeAt(int i) {

        int s = --size;//先得到末尾元素下标,同时将size减1

        int value=queue[i];
        int v = queue[s];
        queue[s] = null;
        if (s == i) {//删除的刚好是末尾这个元素
            return value;
        }

        //将最后一个元素放置在待删除的位置
        queue[i] = v;

        //下调操作
        siftDown(i);

        return value;

    }

private void siftDown(int k) {
        int half = size >>> 1;
        while (k < half) {
            //跟子节点比较

            int least= k * 2 + 1;//先假设较小节点是左节点
            int c=queue[least];
            int right=least+1;
            if (right< size && queue[right]<c){
                c=queue[right];
                least=right;
            }

            if (queue[k]<c){
                break;
            }

            //交换
            queue[least]=queue[k];
            queue[k]=c;
            k=least;

        }
    }

遍历操作

直接遍历数组即可。

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

推荐阅读更多精彩内容