acwing 838. 堆排序 (手写一个堆)

如何手写一个堆？

插入一个数
求集合中最小值
删除最小值
删除任意一个元素
修改任意一个元素
[注]：后面两个是stl 无法直接实现的， stl 的堆就是优先队列

堆
堆是一个完全二叉树（最后一层节点是左右依次排布的）

小根堆为例
每一个节点都小于等于父节点，根节点就是最小值。

实现：

一维数组实现
有两个操作 down(x) 、up(x)
顾名思义：把节点往上移或者往下移动

down(x) ：如果把一个点的值变大了，就需要向下移动；跟叶子节点比较

up(x) ：如果把一个点的值变小了，就需要向上移动；跟父节点比较

如何通过这两个操作实现前面的5个因素呢 heap 表示堆，size表示大小

heap[++size] = x; up(size);
heap[1]
heap[1] = heap[size]; size --; down(1); // 最小值不好直接删除，需要用最后一个值覆盖，然后调整堆
heap[k] = heap[size]; size --; down(k); up(k); // 只会执行1个操作
heap[k] = x; down(k), up(k);

题目

输入一个长度为 n 的整数数列，从小到大输出前 m 小的数。

输入格式
第一行包含整数 n 和 m。

第二行包含 n 个整数，表示整数数列。

输出格式
共一行，包含 m 个整数，表示整数数列中前 m 小的数。

数据范围
1≤m≤n≤105，
1≤数列中元素≤109
输入样例：
5 3
4 5 1 3 2
输出样例：
1 2 3

思路：
只涉及排序，分析一下就用到了操作中的2、3 只涉及到down 操作

code:

#include <iostream>
using namespace std;const int N =  100010;
int n,m;
int h[N], cnt;void down(int u)
{// 核心操作就是看根节点、左儿子、右儿子，哪个是最小值int t = u;  // 用t保存 最小的数字if(u * 2 <= cnt && h[u * 2] < h[t])  t = u * 2;   // 保证左儿子存在， 并且左儿子小于父亲if(u *2 + 1<= cnt && h[u * 2 + 1] < h[t] ) t = u * 2 + 1; // 保证右儿子存在， 并且右儿子小于当前的小的点if(u != t){swap(h[t], h[u]);  // 交换使得最小值在上面down(t);  // 递归处理}
}void up(int u)
{while(u / 2 && h[u / 2] > h[u]) // 如果存在父节点且父节点大{swap(h[u / 2], h[u]);  // 交换u /= 2;     // 递归处理}
}int main()
{cin >> n >> m;  for(int i = 1; i<=n; i++) cin >> h[i];cnt = n;// 初始化堆，从n / 2开始, 能使时间复杂度为O(n), 直接从n开始就会是O(logn)for(int i = n/2; i; i--) down(i);      while(m--)  // 取出前m个小的数字{cout << h[1] << ' ';   // 取出小根堆的最小数字h[1] = h[cnt];  // 将堆底元素（最后一个）放在第一位置cnt --;         // 总的个数 - 1down(1);        // 向下调整堆，维护使得最上面元素是最小的元素}return 0;
}

acwing 838. 堆排序 (手写一个堆)相关推荐

大根堆与小根堆的理解，如何手写一个堆，以及什么时候用自己手写的堆，什么时候用语言提供堆的api，（二者的区别）
大根堆与小根堆的理解,如何手写一个堆,以及什么时候用自己手写的堆,什么时候用语言提供堆的api,(二者的区别) 定义 Heap是一种数据结构具有以下的特点: 1)完全二叉树: 2)heap中存储的值是 ...
来给朕手写一个OOM异常的栗子
2.4 实战 OutOfMemoryError 异常面试官:项目中你有没有遇到过Java虚拟机方面的问题,做过虚拟机调优嘛,来,给朕手写一个堆内存溢出的Demo 注:IDEA内存映像分析工具: ht ...
AcWing 838. 堆排序
原题链接如下: AcWing 838. 堆排序对堆的具体介绍,本人水平和时间都有限,就不在这里赘述了,主要是把从y总那里学来的优雅的手写堆排序的代码复现一下,以加深自己的理解 #include< ...
硬核！手写一个优先队列
文章收录在首发公众号:bigsai 期待你的到访! 前言事情还要从一个故事讲起: 对于上面那只可爱的小狗狗不会,本篇即为该教程,首先,我要告诉这只可爱的小狗狗,这种问题你要使用的数据结构为优先队列, ...
AcWing 838.堆排序
原题链接: AcWing 838. 堆排序输入一个长度为 n 的整数数列,从小到大输出前 m 小的数. 输入格式第一行包含整数 n 和 m. 第二行包含 n 个整数,表示整数数列. 输出格式共一 ...
vue 使用fs_模仿vue-cli，手写一个脚手架
vue-cli 在vue的开发的过程中,经常会使用到vue-cli脚手架工具去生成一个项目.在终端运行命令vue create hello-world后,就会有许多自动的脚本运行. 为什么会这样运行呢 ...
第二季：5公平锁/非公平锁/可重入锁/递归锁/自旋锁谈谈你的理解？请手写一个自旋锁【Java面试题】
第二季:5值传递和引用传递[Java面试题] 前言推荐值传递说明题目 24 TransferValue醒脑小练习第二季:5公平锁/非公平锁/可重入锁/递归锁/自旋锁谈谈你的理解?请手写一个自 ...
手写一个基于NIO的迷你版Tomcat
笔者也建立的自己的公众号啦,平时会分享一些编程知识,欢迎各位大佬支持~ 扫码或微信搜索北风IT之路关注本文公众号地址:手写一个基于NIO的迷你版Tomcat 在很久之前看到了一篇文章写一个迷你版的T ...
手写一个promise用法_手写一个 Promise
1 js 的基本数据类型? 2 JavaScript 有几种类型的值? 3 什么是堆?什么是栈?它们之间有什么区别和联系? 4 内部属性 [Class] 是什么? 5 介绍 js 有哪些内置对象? 6 ...

acwing 838. 堆排序 (手写一个堆)

题目

acwing 838. 堆排序 (手写一个堆)相关推荐

最新文章

热门文章