数据结构——堆

堆的概念

  • 堆(heap)是计算机科学中一类特殊的数据结构的统称。堆通常是一个可以被看做一棵树的数组对象,即是一种顺序储存结构的完全二叉树[1]

提示:完全二叉树

  • 完全二叉树:对一棵深度为k、有n个结点二叉树编号后,各节点的编号与深度为k的满二叉树相同位置的结点的编号相同,这颗二叉树就被称为完全二叉树[2]

堆的性质

  • 堆中某个结点的值总是不大于或不小于其父结点的值
  • 堆总是一棵完全二叉树
  • 除了根结点和最后一个左子结点可以没有兄弟结点,其他结点必须有兄弟结点

最大堆最小堆

  • 最大堆[3]:根结点的键值是所有堆结点键值中最大者,且每个结点的值都比其孩子的值大。

  • 最小堆[3]:根结点的键值是所有堆结点键值中最小者,且每个结点的值都比其孩子的值小。

代码

定义

有限数组形式

int Heap[1024];    //顺序结构的二叉树

若某结点编号为i,且存在左儿子和右儿子,则他们分别对应

Heap[i*2+1];      //左儿子
Heap[i*2+2];      //右儿子

其父节点

Heap[i/2];      //i的父节点

动态数组形式

在项目开发中,经常以动态数组形式出现,在本文主要对这种形式进行介绍

//默认容量
#define DEFAULT_CAPCITY 128

typedef struct _Heap
{
    int *arr;       //储存元素的动态数组
    int size;       //元素个数
    int capacity;   //当前存储的容量   
}Heap;

操作

  • 只使用InitHeap()函数进行初始化即可,AdjustDown()与BuildHeap()仅为堆建立时的内部调用

向下调整结点

  • 以创建最大堆为例
  • 将“判断最大子结点是否大于当前父结点”处的>=改为<=即可创建最小堆
//向下调整,将当前结点与其子结点调整为最大堆
//用static修饰禁止外部调用
static void AdjustDown(Heap& heap, int index)
{
    int cur = heap.arr[index];  //当前待调整结点
    int parent, child;

    /*
        判断是否存在子结点大于当前结点。
        若不存在,堆是平衡的,则不调整;
        若存在,则与最大子结点与之交换,交换后该子节点若还有子结点,则以此方法继续调整。
    */
    for (parent = index; (parent * 2 + 1) < heap.size; parent = child)
    {
        child = parent * 2 + 1; //左子结点

        //取两个子结点中最大节点,(child+1)<heap.size防止越界
        if (((child + 1) < heap.size && (heap.arr[child] < heap.arr[child + 1])))
            child++;

        //判断最大子结点是否大于当前父结点
        if (cur >= heap.arr[child]) //将此处的>=改成<=可构建最小堆
        {
            //不大于,不需要调整
            break;
        }
        else
        {
            //大于,则交换
            heap.arr[parent] = heap.arr[child];
            heap.arr[child] = cur;
        }

    }


}

建立堆

//建立堆,用static修饰禁止外部调用
static void BuildHeap(Heap& heap)
{
    int i;
    //从倒数第二层开始调整(若只有一层则从该层开始)
    for (i = heap.size / 2 - 1; i >= 0; i--)
    {
        AdjustDown(heap, i);
    }
}

初始化

//初始化堆
//参数:被初始化的堆,存放初始数据的数组, 数组大小
bool InitHeap(Heap &heap, int *orginal, int size)
{
    //若容量大于size,则使用默认量,否则为size
    int capacity = DEFAULT_CAPCITY>size?DEFAULT_CAPCITY:size;
    
    heap.arr = new int[capacity];   //分配内存,类型根据实际需要可修改
    if(!heap.arr) return false;     //内存分配失败则返回False
    
    heap.capacity = capacity;       //容量
    heap.size = 0;                  //元素个数置为0
    
    //若存在原始数组则构建堆
    if(size>0)
    {
        /*
        内存拷贝,将orginal的元素拷贝到堆数组arr中
        size*sizeof(int)为元素所占内存空间大小
        */
        memcpy(heap.arr,orginal, size*sizeof(int));
        heap.size = size;   //调整大小
        BuildHeap(heap);    //建堆
    }
    
    return true;
}

打印堆

  • 实际上是一个层序遍历[4]
//以树状的形式打印堆
void PrintHeapAsTreeStyle(Heap hp)
{
    queue<int> que;
    int r = 0;
    que.push(r);
    queue<int> temp;

    while (!que.empty())
    {
        r = que.front();
        que.pop();

        if (r * 2 + 1 < hp.size) temp.push(r * 2 + 1);
        if (r * 2 + 2 < hp.size) temp.push(r * 2 + 2);

        if (que.empty())
        {
            cout << hp.arr[r] << endl;
            que = temp;
            temp = queue<int>();
        }
        else
            cout << hp.arr[r] << " ";

    }

}

测试

main函数

int main()
{
    Heap hp;
    int vals[] = { 1,2,3,87,93,82,92,86,95 };

    if (!InitHeap(hp, vals, sizeof(vals) / sizeof(vals[0])))
    {
        fprintf(stderr, "初始化堆失败!\n");
        exit(-1);
    }

    PrintHeapAsTreeStyle(hp);

    return 0;
}

结果

95
93 92
87 1 82 3
86 2

完整代码

#include <iostream>
#include <queue>

using namespace std;

//默认容量
#define DEFAULT_CAPCITY 128
typedef struct _Heap {
    int* arr;
    int size;
    int capacity;
}Heap;

//向下调整,将当前结点与其子结点调整为最大堆
static void AdjustDown(Heap& heap, int index)
{
    int cur = heap.arr[index];  //当前待调整结点
    int parent, child;

    /*
        判断是否存在子结点大于当前结点。
        若不存在,堆是平衡的,则不调整;
        若存在,则与最大子结点与之交换,交换后该子节点若还有子结点,则以此方法继续调整。
    */
    for (parent = index; (parent * 2 + 1) < heap.size; parent = child)
    {
        child = parent * 2 + 1; //左子结点

        //取两个子结点中最大节点,(child+1)<heap.size防止越界
        if (((child + 1) < heap.size && (heap.arr[child] < heap.arr[child + 1])))
            child++;

        //判断最大子结点是否大于当前父结点
        if (cur >= heap.arr[child]) //将此处的>=改成<=可构建最小堆
        {
            //不大于,不需要调整
            break;
        }
        else
        {
            //大于,则交换
            heap.arr[parent] = heap.arr[child];
            heap.arr[child] = cur;
        }

    }


}

//建立堆,用static修饰禁止外部调用
static void BuildHeap(Heap& heap)
{
    int i;
    //从倒数第二层开始调整(若只有一层则从该层开始)
    for (i = heap.size / 2 - 1; i >= 0; i--)
    {
        AdjustDown(heap, i);
    }
}

//初始化堆
//参数:被初始化的堆,存放初始数据的数组, 数组大小
bool InitHeap(Heap& heap, int* orginal, int size)
{
    //若容量大于size,则使用默认量,否则为size
    int capacity = DEFAULT_CAPCITY > size ? DEFAULT_CAPCITY : size;

    heap.arr = new int[capacity];   //分配内存,类型根据实际需要可修改
    if (!heap.arr) return false;        //内存分配失败则返回False

    heap.capacity = capacity;       //容量
    heap.size = 0;                  //元素个数置为0

    //若存在原始数组则构建堆
    if (size > 0)
    {
        /*
        内存拷贝,将orginal的元素拷贝到堆数组arr中
        size*sizeof(int)为元素所占内存空间大小
        */
        memcpy(heap.arr, orginal, size * sizeof(int));
        heap.size = size;   //调整大小
        BuildHeap(heap);    //建堆
    }

    return true;
}

//以树状的形式打印堆
void PrintHeapAsTreeStyle(Heap hp)
{
    queue<int> que;
    int r = 0;
    que.push(r);
    queue<int> temp;

    while (!que.empty())
    {
        r = que.front();
        que.pop();

        if (r * 2 + 1 < hp.size) temp.push(r * 2 + 1);
        if (r * 2 + 2 < hp.size) temp.push(r * 2 + 2);

        if (que.empty())
        {
            cout << hp.arr[r] << endl;
            que = temp;
            temp = queue<int>();
        }
        else
            cout << hp.arr[r] << " ";

    }

}

int main()
{
    Heap hp;
    int vals[] = { 1,2,3,87,93,82,92,86,95 };

    if (!InitHeap(hp, vals, sizeof(vals) / sizeof(vals[0])))
    {
        fprintf(stderr, "初始化堆失败!\n");
        exit(-1);
    }

    PrintHeapAsTreeStyle(hp);

    return 0;
}

参考


  1. 堆(数据结构)_百度百科

  2. 二叉树 - 菜缤的世界 CairBin's Blog

  3. 最大堆和最小堆_varyall的博客-CSDN博客

  4. 二叉树及其遍历 - 菜缤的世界 CairBin's Blog

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 143,809评论 1 304
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 61,651评论 1 257
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 95,178评论 0 213
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 41,241评论 0 181
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 49,047评论 1 259
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 38,899评论 1 178
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 30,503评论 2 274
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 29,249评论 0 168
  • 想象着我的养父在大火中拼命挣扎,窒息,最后皮肤化为焦炭。我心中就已经是抑制不住地欢快,这就叫做以其人之道,还治其人...
    爱写小说的胖达阅读 29,125评论 6 235
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 32,605评论 0 213
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 29,368评论 2 215
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 30,723评论 1 232
  • 白月光回国,霸总把我这个替身辞退。还一脸阴沉的警告我。[不要出现在思思面前, 不然我有一百种方法让你生不如死。]我...
    爱写小说的胖达阅读 24,285评论 0 32
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 27,190评论 2 214
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 31,634评论 3 209
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 25,651评论 0 9
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,052评论 0 167
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 33,638评论 2 232
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 33,760评论 2 237

推荐阅读更多精彩内容

  • 什么是堆 堆其实就是一棵完全二叉树,即若设二叉树的深度为h,除第 h 层外,其它各层 (1~h-1) 的结点数都达...
    赵阳_c149阅读 718评论 0 1
  • 定义 优先队列:一种特殊的队列,队列中元素出栈的顺序是按照元素的优先权大小,而不是元素入队的先后顺序。 堆的特性:...
    IAM四十二阅读 726评论 1 2
  • 导语 堆的逻辑数据结构实际上是一个可以使用数组实现的完全二叉树(堆也一定是平衡二叉树),所以学习堆,完全二叉树不是...
    曦夫阅读 1,176评论 0 0
  • 一、定义 堆:实际上是一颗 完全二叉树 ,但是它还满足父结点大于(或小于)子结点特性。父结点大于子结点称为最大堆(...
    MangK阅读 2,846评论 0 4
  • 通常情况下我们把堆看成是一棵完全二叉树。堆一般分为两种,一种是最大堆,一种是最小堆。最大堆要求根节点的值即大于左子...
    数据结构和算法阅读 335评论 0 1