• 首页 首页 icon
  • 工具库 工具库 icon
    • IP查询 IP查询 icon
  • 内容库 内容库 icon
    • 快讯库 快讯库 icon
    • 精品库 精品库 icon
    • 问答库 问答库 icon
  • 更多 更多 icon
    • 服务条款 服务条款 icon

一篇解建堆,堆的实现,堆排序,TopK问题C语言《数据结构和算法》

武飞扬头像
原来45
帮助1

目录 

1. 堆的概念及结构 🚀

2. 堆的实现 🚀

2.1. 堆向下调整算法 🚀

2.2. 堆的创建 🚀

2.4. 堆的插入 🚀

2.5. 堆的删除 🚀

3. 堆的应用 🚀

3.1 堆排序 🚀

3.2 TOP-K问题 🚀

注意 

4. 源代码 🚀

4.1. Heap.h 🚀

4.2. Heap.c 🚀

4.3. test.c 🚀


hello✨,大家好呀,这里是原来💖 💛💙随着文章篇幅越来越多可能有很多小伙伴们找不到自己想要看的文章🎁 🎁 🎁C语言入门篇进阶篇深剖篇都整理在这里了哈。然后这里个人主页,比点头像更好找文章哈。🎈 🎈🎈 


期待hxd的支持哈🎉 🎉 🎉

这里附带作者qq:2777137742

                      vx:q2777137742

因为私信消息很多不容易看见哈


最后是打鸡血环节:改变的确很难,但结果值得冒险,拿出点勇气来。路还很长,现在才刚开始而已。过去无可挽回,未来可以改变。🚀 🚀 🚀

1. 堆的概念及结构 🚀

学新通
还有堆的左右子树都是大堆或者小堆,堆顶是最大值或者最小值。

学新通

2. 堆的实现 🚀

在实现堆有两个比较重要的事情就是理解向上调整算法和向下调整算法。

堆的向上调整算法:是为了在插入数据的时候使原来的结构不变,还是一个堆。

堆的向下调整算法:1.是为了建堆。或者给你一个数组这个数组逻辑上可以看做一颗完全二叉树,但是还不是一个堆,我们要通过算法,把它构建成一个堆。2.删除堆数据的时候要用向下调整算法。

2.1. 堆向下调整算法 🚀

现在我们给出一个数组,逻辑上看做一颗完全二叉树。我们通过从根节点开始的向下调整算法可以把它调整成一个小堆。向下调整算法有一个 前提 :左右子树必须是一个堆,才能调整。
int array[] = {27,15,19,18,28,34,65,49,25,37};

学新通

2.2. 堆的创建 🚀

下面我们给出一个数组,这个数组逻辑上可以看做一颗完全二叉树,但是还不是一个堆,现在我们通过向下调整算法,把它构建成一个堆。根节点左右子树不是堆,我们怎么调整呢?这里我们从倒数的第一个非叶子节点子树开始调整,一直调整到根节点的树,就可以调整成堆。
int a[] = {1,5,3,8,7,6};
学新通

2.3. 建堆时间复杂度 🚀

因为堆是完全二叉树,而满二叉树也是完全二叉树,此处为了简化使用满二叉树来证明 ( 时间复杂度本来看的就是近似值,多几个节点不影响最终结果)

学新通

2.4. 堆的插入 🚀

先插入一个10到数组的尾上,再进行向上调整算法,直到满足堆。(数据从下往上调,所以是向上调整算法)

学新通

2.5. 堆的删除 🚀

删除堆是删除堆顶的数据,将堆顶的数据根最后一个数据一换,然后删除数组最后一个数据,再进行向下调整算法。
学新通

3. 堆的应用 🚀

3.1 堆排序 🚀

堆排序即利用堆的思想来进行排序,总共分为两个步骤:
1. 建堆
升序:建大堆
降序:建小堆
2. 利用堆删除思想来进行排序(时间复杂度:NLogN 和 qsort一个等级哈)
建堆和堆删除中都用到了向下调整,因此掌握了向下调整,就可以完成堆排序。
学新通

学新通

3.2 TOP-K问题 🚀

TOP-K 问题:即求数据结合中前 K 个最大的元素或者最小的元素,一般情况下数据量都比较大
比如:专业前 10 名、世界 500 强、富豪榜、游戏中前 100的活跃玩家等。
对于 Top-K 问题,能想到的最简单直接的方式就是排序,但是:如果数据量非常大,排序就不太可取了 ( 可能数据都不能一下子(指时间短暂或动作迅速)(指时间短暂或动作迅速)全部加载到内存中) 。最佳的方式就是用堆来解决,基本思路如下:
1. 用数据集合中前 K 个元素来建堆
k 个最大的元素,则建小堆
k 个最小的元素,则建大堆
2. 用剩余的 N-K 个元素依次与堆顶元素来比较,不满足则替换堆顶元素 

注意 

上面说的所有,如果不太理解,我们直接看下面的代码,作者是认为一个程序员代码可能给你的理解要比大白话要更直接哈。

4. 源代码 🚀

这里作者是建的小堆,如果要建大堆就改作者在代码里面声明的地方,就把三个小于改成大于就可以了哈。

源代码里面解决了:建堆(给一个数组建堆),堆的各项操作,堆排序,TopK问题。 

PS:代码里面有详细的解释哈。

main函数里面test1里面是建堆的各项操作的测试及堆排序的测试。

                      test2里面是给一个数组然后建堆的测试哈。

4.1. Heap.h 🚀

  1.  
    #define _CRT_SECURE_NO_WARNINGS 1
  2.  
     
  3.  
    #include<stdio.h>
  4.  
    #include<stdlib.h>
  5.  
    #include<assert.h>
  6.  
    #include<string.h>
  7.  
    #include<errno.h>
  8.  
    #include<stdbool.h>
  9.  
    #include<time.h>
  10.  
     
  11.  
    typedef int HPDataType;
  12.  
    typedef struct Heap
  13.  
    {
  14.  
    HPDataType* a;
  15.  
    size_t size;
  16.  
    size_t capacity;
  17.  
    }Heap;
  18.  
     
  19.  
    //初始化
  20.  
    void HeapInit(Heap* php);
  21.  
    //打印
  22.  
    void HeapPrint(Heap* php);
  23.  
    // 堆的构建
  24.  
    void HeapCreate(Heap* php, HPDataType* a, int n);
  25.  
    // 堆的销毁
  26.  
    void HeapDestory(Heap* php);
  27.  
    // 堆的插入
  28.  
    void HeapPush(Heap* php, HPDataType x);
  29.  
    // 堆的删除
  30.  
    void HeapPop(Heap* php);
  31.  
    // 取堆顶的数据
  32.  
    HPDataType HeapTop(Heap* php);
  33.  
    // 堆的数据个数
  34.  
    size_t HeapSize(Heap* php);
  35.  
    // 堆的判空
  36.  
    bool HeapEmpty(Heap* php);
  37.  
     
  38.  
    //堆排序 NLogN
  39.  
    void HeapSort(int* a, size_t size);
  40.  
     
  41.  
     
  42.  
    // TopK问题:找出N个数里面最大/最小的前K个问题。
  43.  
    // 比如:未央区排名前10的泡馍,西安交通大学王者荣耀排名前10的韩信,全国排名前10的李白。等等问题都是Topk问题,
  44.  
    // 需要注意:
  45.  
    // 找最大的前K个,建立K个数的小堆
  46.  
    // 找最小的前K个,建立K个数的大堆
  47.  
    void PrintTopK(int* a, int n, int k);
  48.  
    void TestTopk();

4.2. Heap.c 🚀

  1.  
    #define _CRT_SECURE_NO_WARNINGS 1
  2.  
     
  3.  
    #include"Heap.h"
  4.  
     
  5.  
     
  6.  
    void AdjustDown(HPDataType* a, size_t sz, size_t root);
  7.  
     
  8.  
    堆的构建
  9.  
    //void HeapCreate(Heap* php, HPDataType* a, int n)
  10.  
    //{
  11.  
    // assert(php);
  12.  
    // HPDataType* new = (HPDataType*)malloc(sizeof(HPDataType) * n);
  13.  
    // if (new == NULL)
  14.  
    // {
  15.  
    // perror("HeapCreate failed!\n");
  16.  
    // exit(-1);
  17.  
    // }
  18.  
    // php->a = new;
  19.  
    // php->capacity=n;
  20.  
    // php->size = 0;
  21.  
    // int curpos = php->size / 2 - 1;
  22.  
    // while (curpos >= 0)
  23.  
    // {
  24.  
    // AdjustDown(php->a, 7,curpos);
  25.  
    // curpos--;
  26.  
    // }
  27.  
    //}
  28.  
     
  29.  
    //给数组建堆
  30.  
    void HeapCreate(Heap* php, int arr[], int n)
  31.  
    {
  32.  
    php->a = (int*)malloc(sizeof(HPDataType) * n);
  33.  
    assert(php->a != NULL);
  34.  
    php->capacity = n;
  35.  
    php->size = n;
  36.  
     
  37.  
    //先把数据进堆
  38.  
    for (int i = 0; i < n; i)
  39.  
    php->a[i] = arr[i];
  40.  
     
  41.  
    //先找到最后一个分支的第一个非叶子节点
  42.  
    int curpos = php->size / 2 - 1;
  43.  
    while (curpos >= 0)
  44.  
    {
  45.  
    AdjustDown(php->a, 7,curpos);
  46.  
    curpos--;
  47.  
    }
  48.  
     
  49.  
    }
  50.  
     
  51.  
    //初始化
  52.  
    void HeapInit(Heap* php)
  53.  
    {
  54.  
    assert(php);
  55.  
    php->a = NULL;
  56.  
    php->capacity = 0;
  57.  
    php->size = 0;
  58.  
    }
  59.  
     
  60.  
    //打印
  61.  
    void HeapPrint(Heap* php)
  62.  
    {
  63.  
    assert(php);
  64.  
    assert(php->size > 0);
  65.  
    for (size_t i = 0; i < php->size; i )
  66.  
    {
  67.  
    printf("%d ", php->a[i]);
  68.  
    }
  69.  
    printf("\n");
  70.  
    }
  71.  
     
  72.  
    // 堆的销毁
  73.  
    void HeapDestory(Heap* php)
  74.  
    {
  75.  
    assert(php);
  76.  
    php->a = NULL;
  77.  
    php->capacity = 0;
  78.  
    php->size = 0;
  79.  
    free(php->a);
  80.  
    }
  81.  
     
  82.  
    //交换
  83.  
    void Swap(HPDataType* p1, HPDataType* p2)
  84.  
    {
  85.  
    HPDataType tmp = *p1;
  86.  
    *p1 = *p2;
  87.  
    *p2 = tmp;
  88.  
    }
  89.  
     
  90.  
    //小堆为例 大堆换符合就可以了
  91.  
    //向上调整算法
  92.  
    void AdjustUP(HPDataType* a, size_t child)
  93.  
    {
  94.  
    size_t parent = (child - 1) / 2;
  95.  
    while (child>0)
  96.  
    {
  97.  
    if (a[child] < a[parent])
  98.  
    {
  99.  
    Swap(&a[child], &a[parent]);
  100.  
    }
  101.  
    else
  102.  
    {
  103.  
    break;
  104.  
    }
  105.  
    child = parent;
  106.  
    parent= (child - 1) / 2;
  107.  
    }
  108.  
    }
  109.  
     
  110.  
    //向下调整算法
  111.  
    void AdjustDown(HPDataType* a, size_t sz, size_t root)
  112.  
    {
  113.  
    //先找左右小的孩子
  114.  
    //再和root的比 比root小就交换
  115.  
    //root等于那个要交换的孩子 孩子再选 然后迭代
  116.  
     
  117.  
    //改为这个可能更好看一点
  118.  
    //size_t parent = root;
  119.  
    //这里默认左孩子
  120.  
    //size_t child = parent * 2 1;
  121.  
     
  122.  
    //这里默认左孩子
  123.  
    size_t child = root * 2 1;
  124.  
    //sz的作用就是确保左右还在在数组内
  125.  
    while (child<sz)
  126.  
    {
  127.  
    //小心右孩子不存在
  128.  
    //找左右孩子小的
  129.  
    //还有注意要把比较左右孩子放进循环,因为每次都要比较
  130.  
    if (child 1<sz && a[child 1] < a[child])
  131.  
    {
  132.  
    child = child 1;
  133.  
    }
  134.  
     
  135.  
    if (a[child] < a[root])
  136.  
    {
  137.  
    Swap(&a[child], &a[root]);
  138.  
    root = child;
  139.  
    child = root * 2 1;
  140.  
    }
  141.  
    else
  142.  
    {
  143.  
    break;
  144.  
    }
  145.  
    }
  146.  
     
  147.  
    }
  148.  
     
  149.  
    // 堆的插入
  150.  
    void HeapPush(Heap* php, HPDataType x)
  151.  
    {
  152.  
    assert(php);
  153.  
    if (php->capacity == php->size)
  154.  
    {
  155.  
    size_t newcapacity = php->capacity == 0 ? 4 : php->capacity*2;
  156.  
    HPDataType* new = (HPDataType*)realloc(php->a, sizeof(HPDataType) * newcapacity);
  157.  
    if (new == NULL)
  158.  
    {
  159.  
    printf("%s", strerror(errno));
  160.  
    exit(-1);
  161.  
    }
  162.  
    php->a = new;
  163.  
    php->capacity = newcapacity;
  164.  
    }
  165.  
    php->a[php->size] = x;
  166.  
    php->size ;
  167.  
    //上面是插入数据
  168.  
    //下面是 向上调整算法 使插入数据后还是一个小堆
  169.  
    AdjustUP(php->a,php->size-1);
  170.  
    }
  171.  
     
  172.  
    // 堆的删除
  173.  
    void HeapPop(Heap* php)
  174.  
    {
  175.  
    assert(php);
  176.  
    assert(php->size > 0);
  177.  
    Swap(&php->a[0], &php->a[php->size - 1]);
  178.  
    php->size--;
  179.  
    //传参注意一下
  180.  
    AdjustDown(php->a,php->size,0);
  181.  
    }
  182.  
     
  183.  
    // 取堆顶的数据
  184.  
    HPDataType HeapTop(Heap* php)
  185.  
    {
  186.  
    assert(php);
  187.  
    assert(php->size > 0);
  188.  
    HPDataType top = php->a[0];
  189.  
    return top;
  190.  
    }
  191.  
     
  192.  
    // 堆的判空
  193.  
    bool HeapEmpty(Heap* php)
  194.  
    {
  195.  
    assert(php);
  196.  
    return php->size == 0;
  197.  
    }
  198.  
     
  199.  
    // 堆的数据个数
  200.  
    size_t HeapSize(Heap* php)
  201.  
    {
  202.  
    assert(php);
  203.  
    return php->size;
  204.  
    }
  205.  
     
  206.  
    //现在默认小堆,想变为大堆就把上面的向上向下调整算法改一下判断就可以了(3个)
  207.  
    //堆排序时间复杂度: NLogN
  208.  
    //为什么是NLogN:因为每次插入数据都是一层只插入一个数据(假设是满二叉树,总
  209.  
    // 节点个数就是2^k-1:(2^(k-1)*2-1/(2-1))=N),k=Log2(k 1),即LogN)
  210.  
    // 又因为有N个数要排
  211.  
    //所以就是N*LogN
  212.  
    void HeapSort(int* a, size_t size)
  213.  
    {
  214.  
    assert(a);
  215.  
    Heap hp;
  216.  
    HeapInit(&hp);
  217.  
     
  218.  
    size_t i = 0;
  219.  
    for (i = 0; i < size; i )
  220.  
    {
  221.  
    HeapPush(&hp, a[i]);
  222.  
    }
  223.  
     
  224.  
    while (!HeapEmpty(&hp))
  225.  
    {
  226.  
    printf("%d ", HeapTop(&hp));
  227.  
    HeapPop(&hp);
  228.  
    }
  229.  
    HeapDestory(&hp);
  230.  
    }
  231.  
     
  232.  
     
  233.  
    // TopK问题:找出N个数里面最大/最小的前K个问题。
  234.  
    // 比如:未央区排名前10的泡馍,西安交通大学王者荣耀排名前10的韩信,
  235.  
    // 全国排名前10的李白。等等问题都是Topk问题,
  236.  
    // 需要注意:
  237.  
    // 找最大的前K个,建立K个数的小堆
  238.  
    // 找最小的前K个,建立K个数的大堆
  239.  
    void PrintTopK(int* a, int n, int k)
  240.  
    {
  241.  
    assert(a);
  242.  
    assert(k < n);
  243.  
     
  244.  
    Heap hp;
  245.  
    HeapInit(&hp);
  246.  
    int i = 0;
  247.  
    for (i = 0; i < n; i )
  248.  
    {
  249.  
    HeapPush(&hp, a[i]);
  250.  
    }
  251.  
     
  252.  
    while (k--)
  253.  
    {
  254.  
    printf("%d ", HeapTop(&hp));
  255.  
    HeapPop(&hp);
  256.  
    }
  257.  
    }
  258.  
     
  259.  
    void TestTopk()
  260.  
    {
  261.  
    //int arr[] = { 15,18,19,25,28,34,65,49,27,37 };
  262.  
    //int sz = sizeof(arr) / sizeof(arr[0]);
  263.  
    //int k = 5;
  264.  
    //PrintTopK(arr, sz, k);
  265.  
    int n = 10000;
  266.  
    int* a = (int*)malloc(sizeof(int) * n);
  267.  
    srand(time(0));
  268.  
    for (size_t i = 0; i < n; i)
  269.  
    {
  270.  
    a[i] = rand() % 1000000;
  271.  
    }
  272.  
    a[5] = 1000000 1;
  273.  
    a[1231] = 1000000 2;
  274.  
    a[531] = 1000000 3;
  275.  
    a[5121] = 1000000 4;
  276.  
    a[115] = 1000000 5;
  277.  
    a[2335] = 1000000 6;
  278.  
    a[9999] = 1000000 7;
  279.  
    a[76] = 1000000 8;
  280.  
    a[423] = 1000000 9;
  281.  
    a[3144] = 1000000 10;
  282.  
    PrintTopK(a, n, 10);
  283.  
    }

4.3. test.c 🚀

  1.  
    #define _CRT_SECURE_NO_WARNINGS 1
  2.  
     
  3.  
    #include"Heap.h"
  4.  
     
  5.  
    void test1()
  6.  
    {
  7.  
    //Heap hp;
  8.  
    //HeapInit(&hp);
  9.  
     
  10.  
    HeapCreate(&hp, arr, 10);
  11.  
     
  12.  
    //HeapPush(&hp, 1);
  13.  
    //HeapPrint(&hp);
  14.  
     
  15.  
    //HeapPush(&hp, 5);
  16.  
    //HeapPrint(&hp);
  17.  
     
  18.  
    //HeapPush(&hp, 0);
  19.  
    //HeapPrint(&hp);
  20.  
     
  21.  
    //HeapPush(&hp, 8);
  22.  
    //HeapPrint(&hp);
  23.  
     
  24.  
    //HeapPush(&hp, 3);
  25.  
    //HeapPrint(&hp);
  26.  
     
  27.  
    //HeapPush(&hp, 9);
  28.  
    //HeapPrint(&hp);
  29.  
     
  30.  
    //HeapPush(&hp, 17);
  31.  
    //HeapPrint(&hp);
  32.  
     
  33.  
    //HeapPush(&hp, 13);
  34.  
    //HeapPush(&hp, 15);
  35.  
    //HeapPush(&hp, 20);
  36.  
    //HeapPrint(&hp);
  37.  
     
  38.  
    //printf("size=%d \n", HeapSize(&hp));
  39.  
    //while (!HeapEmpty(&hp))
  40.  
    //{
  41.  
    // printf("%d ", HeapTop(&hp));
  42.  
    // HeapPop(&hp);
  43.  
    //}
  44.  
     
  45.  
    //HeapDestory(&hp);
  46.  
     
  47.  
    /// //
  48.  
    //升序
  49.  
    int arr[] = { 15,18,19,25,28,34,65,49,27,37 };
  50.  
    size_t sz = sizeof(arr) / sizeof(arr[0]);
  51.  
    HeapSort(arr, sz);
  52.  
     
  53.  
    //
  54.  
     
  55.  
    TestTopk();
  56.  
    }
  57.  
     
  58.  
    void test2()
  59.  
    {
  60.  
    Heap hp;
  61.  
    HeapInit(&hp);
  62.  
    int arr[] = { 3,7,5,2,9,10,15 };
  63.  
    HeapCreate(&hp, arr, 7);
  64.  
    for (int i = 0; i < 7; i )
  65.  
    {
  66.  
    printf("%d ", hp.a[i]);
  67.  
    }
  68.  
    }
  69.  
     
  70.  
    int main()
  71.  
    {
  72.  
    test1();
  73.  
     
  74.  
    //test2();
  75.  
    return 0;
  76.  
    }

最后的最后,创作不易,希望读者三连支持💖

赠人玫瑰,手有余香💖

这篇好文章是转载于:学新通技术网

  • 版权申明: 本站部分内容来自互联网,仅供学习及演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,请提供相关证据及您的身份证明,我们将在收到邮件后48小时内删除。
  • 本站站名: 学新通技术网
  • 本文地址: /boutique/detail/tanhgfbcbg
系列文章
更多 icon
同类精品
更多 icon
继续加载