数据结构 堆

手写堆,而非stl中的堆

如何手写一个堆?

//将数组建成堆 <O(n)

for (int i = n / 2;i;i--) //从n/2开始down

down(i);

从n/2元素开始down,最下面一层元素的个数是n/2,其余上面的元素的个数是n/2,从最下面一层到最高层,每层元素的个数是n/2^m【m从下到上,从1计算,m=1时是最下面一层】,时间复杂度就是元素个数*高度【高度从下到上,从0计算】

性质

1.堆是一棵完全二叉树【除叶子结点之外,所有结点都是非空的】

2.小根堆:每个点的值都是小于等于其左右两个儿子的,根结点就是整个数据中的最小值

静态数组存储------一维数组

用一个一维数组来存,根结点放在数组开头,1号点是根结点,结点x的左儿子下标为2x,右儿子下标为2x+1

两个基本操作

对于所有的堆操作而言,每个操作都可以使用这两个操作构建。

down(x) 向下调整------从下往上down一次就变成一个堆

大数位于上面,需要向下移

每次与其两个儿子进行比较,找到较小的儿子与其进行交换,直到小于所有的儿子为止【该结点的子结点都大于该结点】

//递归实现

void down(int u) {

int t = u;//用t表示三个点中的最小值

if (u * 2 <= size1 && h[u * 2] < h[t])//先判断是否有左儿子,然后判断左儿子是否小于其本身,如果成立,交换

t = u * 2;

if (u * 2 + 1 <= size1 && h[u * 2 + 1] < h[t])//再判断是否有右儿子,然后判断右儿子是否小于其本身,如果成立,交换

t = u * 2 + 1;

//最终,t存的就是三个点中最小的结点编号

if (u != t) {//如果u!=t,说明根结点就不是最小的,需要交换

swap(h[u], h[t]);//交换

down(t);//递归,交换之前h[t]<=h[u],交换之后h[t]>h[u],h[t]中存的是大数,对其再进行down操作,即递归

}

return;

}

up(x) 向上调整

小数位于下面,需要向上移

每次只需要与其根结点比较,如果小于其根结点,就与其根结点进行交换,直到>=其根结点为止

//循环实现

void up(int u) {

while (u / 2 && h[u / 2] > h[u]) {//u的父结点为u/2,父结点存在且大于本身,交换

swap(h[u / 2], h[u]);

u = u / 2;

}

return;

}

操作

【前三最重要】

【下标从1开始】

1.向集合中插入一个数

在整个堆的最后一个位置插入,然后再向上调整

heap[++size]=x;

up(size);

2.求集合中的最小值

heap[1];

3.删除最小值

用整个堆的最后一个元素覆盖掉堆顶的元素,然后size--,然后向下调整

因为删去最后一个结点特别容易,而删除根结点却不易

heap[1]=heap[size];

size--;

down(1);

4.删除任意一个元素

用堆的最后一个结点覆盖该结点,然后size--,然后向下调整(变大)、向上调整(变小),二选一执行

heap[k]=heap[size];

size--;

down(k);//变大

up(k);//变小

5.修改任意一个操作

heap[k]=x;

down(k);//变大

up(k);//变小

例题------堆排序

题目描述

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

输入格式

第一行包含整数n和m。

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

输出格式

共一行,包含m个整数,表示整数数列中前m小的数。

数据范围

1≤m≤n≤10^5,

1≤数列中元素≤10^9

输入样例

5 3

4 5 1 3 2

输出样例

1 2 3

#include<iostream>

#include<algorithm>

using namespace std;

const int N = 100010;

int n, m;

int h[N], size1;//h[N]就是heap[N],size1存储当前有多少个元素

int main() {

scanf("%d%d", &n, &m);

for (int i = 1;i <= n;i++)

scanf("%d", &h[i]);

size1 = n;

//将数组建成堆 <O(n)

for (int i = n / 2;i;i--) //从n/2开始down

down(i);

while (m--) {

printf("%d ", h[1]);//每次输出堆顶元素,并将其删去

h[1] = h[size1];

size1--;

down(1);

}

return 0;

}

例题------模拟堆[包含映射]

增加两个数组

使用两个数组维护两个映射关系,ph[k] 存第k个插入的点在堆中的下标,hp[k] 存堆中下标为k的点是第几个插入的点

增加的原因

因为按第几个插入元素更改内容,需要知道第i个插入的元素在堆中的下标,所以需要ph的存在,而因为元素在进行down与up操作时,使得ph内容与实际堆的元素不对应,所以要改变ph,而改变ph应该知道,每一个下标对应的插入元素是第几个,所以需要hp的存在。每次交换位置时应该共同维护这两个数组。

交换操作

void heap_swap(int a, int b) {

swap(ph[hp[a]], ph[hp[b]]);//交换指向

swap(hp[a], hp[b]);

swap(h[a], h[b]);

return;

}

交换堆中的两个元素时,hp 和 ph 也改变。先改变 hp 和 ph 中的内容,然后改变这两个结点中的值。先根据交换的下标找到对应的 hp,并以两个 hp 元素值作为 ph 的下标,交换这2个 ph 元素值。之后根据下标交换 hp。

swap(ph[hp[a]], ph[hp[b]]);为什么这里的ph的下标是hp的元素值而不是堆中元素的编号?

因为ph的下标k的含义【即ph[k]的k的含义】是第k个插入的点,所以我们要找到第k个插入的点而不是堆中下标为k的点。

改变后的操作

up 操作

手写的heap_swap函数代替原来的swap函数

void up(int u) {

while (u / 2 && h[u / 2] > h[u]) {//u的父结点为u/2,父结点存在且大于本身,交换

heap_swap(u / 2, u);

u = u / 2;

}

return;

}

down 操作

手写的heap_swap函数代替原来的swap函数

void down(int u) {

int t = u;//用t表示三个点中的最小值

if (u * 2 <= size1 && h[u * 2] < h[t])//先判断是否有左儿子,然后判断左儿子是否小于其本身,如果成立,交换

t = u * 2;

if (u * 2 + 1 <= size1 && h[u * 2 + 1] < h[t])//再判断是否有右儿子,然后判断右儿子是否小于其本身,如果成立,交换

t = u * 2 + 1;

//最终,t存的就是三个点中最小的结点编号

if (u != t) {//如果u!=t,说明根结点就不是最小的,需要交换

heap_swap(u, t);//交换

down(t);

}

return;

}

向集合中插入一个数

添加hp和ph数组中的映射关系

scanf("%d", &x); //向堆中插入x

size1++;

m++;

ph[m] = size1;

hp[size1] = m;

h[size1] = x;

up(size1);

输出集合中的最小值

printf("%d\n", h[1])

删除最小值

手写的heap_swap函数代替原来的swap函数

heap_swap(1, size1);

size1--;

down(1);

删除第k个插入的元素

scanf("%d", &k);//输入k

k = ph[k];//找到第k个插入的元素在堆中的下标,然会对其进行删除

heap_swap(k, size1);//用堆中最后一个元素覆盖找到的元素,然会进行调整

size1--;

down(k), up(k);

修改第k个插入的元素

scanf("%d%d", &k, &x);//输入k和修改后的值x

k = ph[k];//找到第k个插入的元素在堆中的下标,然后修改其值,修改后进行调整

h[k] = x;

down(k), up(k);

题目描述

维护一个集合,初始时集合为空,支持如下几种操作:

"I x",插入一个数x;

"PM",输出当前集合中的最小值;

"DM",删除当前集合中的最小值(当最小值不唯一时,删除最早插入的最小值);

"D k",删除第k个插入的数;

"C k x",修改第k个插入的数,将其变为x;

现在要进行N次操作,对于所有第2个操作,输出当前集合的最小值。

输入格式

第一行包含整数N。

接下来N行,每行包含一个操作指令,操作指令为"I x","PM","DM","D k"或"C k x"中的一种。

输出格式

对于每个输出指令"PM",输出一个结果,表示当前集合中的最小值。

每个结果占一行。

数据范围

1≤N≤10^5

−10^9≤x≤10^9

数据保证合法。

输入样例

8

I -10

PM

I -10

D 1

C 2 8

I 6

PM

DM

输出样例

-10

6

#include<iostream>

#include<algorithm>

#include<string.h>

using namespace std;

const int N = 100010;

int h[N], ph[N], hp[N], size1;//h[N]就是heap[N],size1存储当前有多少个元素,ph[k]存第k个插入数组的下标

//ph[j]=k【第j次插入数组的数的下标是k】,hp[k]=j【下标为k的数是第j次插入数组中的数】

void heap_swap(int a, int b) {

swap(ph[hp[a]], ph[hp[b]]);//交换指向

swap(hp[a], hp[b]);

swap(h[a], h[b]);

return;

}

void down(int u) {

int t = u;//用t表示三个点中的最小值

if (u * 2 <= size1 && h[u * 2] < h[t])//先判断是否有左儿子,然后判断左儿子是否小于其本身,如果成立,交换

t = u * 2;

if (u * 2 + 1 <= size1 && h[u * 2 + 1] < h[t])//再判断是否有右儿子,然后判断右儿子是否小于其本身,如果成立,交换

t = u * 2 + 1;

//最终,t存的就是三个点中最小的结点编号

if (u != t) {//如果u!=t,说明根结点就不是最小的,需要交换

heap_swap(u, t);//交换

down(t);

}

return;

}

void up(int u) {

while (u / 2 && h[u / 2] > h[u]) {//u的父结点为u/2,父结点存在且大于本身,交换

heap_swap(u / 2, u);

u = u / 2;

}

return;

}

int main() {

int n, m = 0;

scanf("%d", &n);

while (n--) {

char op[10];

int k, x;

scanf("%s", op);

if (!strcmp(op, "I")) {

scanf("%d", &x);

size1++;

m++;

ph[m] = size1;

hp[size1] = m;

h[size1] = x;

up(size1);

}

else if (!strcmp(op, "PM"))

printf("%d\n", h[1]);

else if (!strcmp(op, "DM")) {

heap_swap(1, size1);

size1--;

down(1);

}

else if (!strcmp(op, "D")) {

scanf("%d", &k);

k = ph[k];

heap_swap(k, size1);

size1--;

down(k), up(k);

}

else {

scanf("%d%d", &k, &x);

k = ph[k];

h[k] = x;

down(k), up(k);

}

}

return 0;

}

相关推荐
Heisenberg~1 小时前
详解八大排序(五)------(计数排序,时间复杂度)
c语言·数据结构·排序算法
Hera_Yc.H12 小时前
数据结构之一:复杂度
数据结构
肥猪猪爸13 小时前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
linux_carlos13 小时前
环形缓冲区
数据结构
readmancynn13 小时前
二分基本实现
数据结构·算法
Bucai_不才13 小时前
【数据结构】树——链式存储二叉树的基础
数据结构·二叉树
盼海13 小时前
排序算法(四)--快速排序
数据结构·算法·排序算法
一直学习永不止步14 小时前
LeetCode题练习与总结:最长回文串--409
java·数据结构·算法·leetcode·字符串·贪心·哈希表
珹洺15 小时前
C语言数据结构——详细讲解 双链表
c语言·开发语言·网络·数据结构·c++·算法·leetcode
几窗花鸢15 小时前
力扣面试经典 150(下)
数据结构·c++·算法·leetcode