c++算法学习笔记 (19) 堆

简介: c++算法学习笔记 (19) 堆

1.堆排序:

(1)插入一个数:heap[++size]=x;up(size);//在最后插入,再往上移

(2)求集合中最小值:heap[1]

(3)删除最小值:swap(heap[1],heap[size]);size--;down(1);//将最小值移到最后直接删除,再将heap[1]下移到合适位置

(4)删除任意一个元素:swap(heap[k],heap[size]);size--;down(1) or up(1);//down和up选一个执行

(5)修改任意一个元素:heap[k]=x;down(1) or up(1);//down和up选一个执行

例题:

输入一个长度为 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 <bits/stdc++.h>
using namespace std;
const int N = 100010;
int n, m;
int h[N];
int sz;
void down(int u) // 每次往下走
{
    int t = u;
    if (u * 2 <= sz && h[u * 2] < h[t]) // 左儿子比父亲小
    {
        t = u * 2;
    }
    if (u * 2 + 1 <= sz && h[u * 2 + 1] < h[t]) // 右儿子比父亲小
    {
        t = u * 2 + 1;
    }
    if (u != t)
    {
        swap(h[u], h[t]);
        down(t);
    }
}
void up(int u)
{                                    // 每次往上走
    while (u / 2 && h[u / 2] > h[u]) // 只和父亲比较,而down需要和两个儿子比较
    {
        swap(h[u / 2], h[u]);
        u /= 2;
    }
}
int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
        cin >> h[i];
    sz = n;
    for (int i = n / 2; i; i--)
    { // 从n/2(即倒数第二层开始down,到根)
        down(i);
    }
    while (m--)
    {
        cout << h[1] << " ";
        h[1] = h[sz];
        sz--;
        down(1);
    }
}


2.模拟堆

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

  1. I x,插入一个数 x;
  2. PM,输出当前集合中的最小值;
  1. DM,删除当前集合中的最小值(数据保证此时的最小值唯一);
  2. D k,删除第 k 个插入的数;
  3. C k x,修改第 k 个插入的数,将其变为 x;

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

输入格式

第一行包含整数 N。

接下来 N 行,每行包含一个操作指令,操作指令为 I xPMDMD kC k x 中的一种。

输出格式

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

每个结果占一行。

数据范围

1≤N≤10^5

−109≤x≤10^9

数据保证合法。

输入样例:
8
I -10
PM
I -10
D 1
C 2 8
I 6
PM
DM


输出样例:
-10
6


#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
int n, k;
int h[N];
int sz;
int ph[N], hp[N];            // ph[k]存第k个插入的点的下标,hp[k]存堆里下标为k的点是第几个插入的点
void heap_swap(int a, int b) // ab为下标
{
    swap(ph[hp[a]], ph[hp[b]]);
    swap(hp[a], hp[b]);
    swap(h[a], h[b]);
}
 
void down(int u) // 每次往下走
{
    int t = u;
    if (u * 2 <= sz && h[u * 2] < h[t]) // 左儿子比父亲小
    {
        t = u * 2;
    }
    if (u * 2 + 1 <= sz && h[u * 2 + 1] < h[t]) // 右儿子比父亲小
    {
        t = u * 2 + 1;
    }
    if (u != t)
    {
        heap_swap(u, t);
        down(t);
    }
}
void up(int u)
{                                    // 每次往上走
    while (u / 2 && h[u / 2] > h[u]) // 只和父亲比较,而down需要和两个儿子比较
    {
        heap_swap(u / 2, u);
        u /= 2;
    }
}
int main()
{
    cin >> n;
    while (n--)
    {
        char op[10];
        cin >> op;
        if (!strcmp(op, "I"))
        { // 插入
            int x;
            cin >> x;
            sz++;
            k++;        // 存第几个插入的数
            ph[k] = sz; // ph[k]存第k个插入的点的下标
            hp[sz] = k; // hp[k]存堆里下标为k的点是第几个插入的点
            h[sz] = x;
            up(sz);
        }
        else if (!strcmp(op, "PM")) // 输出集合最小值
            cout << h[1] << endl;
        else if (!strcmp(op, "DM")) // 删除集合最小值
        {
            heap_swap(sz, 1);
            sz--;
            down(1);
        }
        else if (!strcmp(op, "D")) // 删除第k个插入的数
        {
            int kk;
            cin >> kk;
            kk = ph[kk]; // 存第k个插入的点的下标
            heap_swap(sz, kk);
            sz--;
            down(kk);
            up(kk); // down和up最多只会执行一个
        }
        else
        { // 将第k个插入的数修改
            int kk, x;
            cin >> kk >> x;
            kk = ph[kk];
            h[kk] = x;
            down(kk);
            up(kk); // down和up最多只会执行一个
        }
    }
    return 0;
}


相关文章
|
18天前
|
存储 监控 算法
员工屏幕监控系统之 C++ 图像差分算法
在现代企业管理中,员工屏幕监控系统至关重要。本文探讨了其中常用的图像差分算法,该算法通过比较相邻两帧图像的像素差异,检测屏幕内容变化,如应用程序切换等。文中提供了C++实现代码,并介绍了其在实时监控、异常行为检测和数据压缩等方面的应用,展示了其实现简单、效率高的特点。
40 15
|
18天前
|
算法 Serverless 数据处理
从集思录可转债数据探秘:Python与C++实现的移动平均算法应用
本文探讨了如何利用移动平均算法分析集思录提供的可转债数据,帮助投资者把握价格趋势。通过Python和C++两种编程语言实现简单移动平均(SMA),展示了数据处理的具体方法。Python代码借助`pandas`库轻松计算5日SMA,而C++代码则通过高效的数据处理展示了SMA的计算过程。集思录平台提供了详尽且及时的可转债数据,助力投资者结合算法与社区讨论,做出更明智的投资决策。掌握这些工具和技术,有助于在复杂多变的金融市场中挖掘更多价值。
46 12
|
9天前
|
存储 监控 算法
公司监控上网软件架构:基于 C++ 链表算法的数据关联机制探讨
在数字化办公时代,公司监控上网软件成为企业管理网络资源和保障信息安全的关键工具。本文深入剖析C++中的链表数据结构及其在该软件中的应用。链表通过节点存储网络访问记录,具备高效插入、删除操作及节省内存的优势,助力企业实时追踪员工上网行为,提升运营效率并降低安全风险。示例代码展示了如何用C++实现链表记录上网行为,并模拟发送至服务器。链表为公司监控上网软件提供了灵活高效的数据管理方式,但实际开发还需考虑安全性、隐私保护等多方面因素。
15 0
公司监控上网软件架构:基于 C++ 链表算法的数据关联机制探讨
|
2月前
|
负载均衡 算法 安全
探秘:基于 C++ 的局域网电脑控制软件自适应指令分发算法
在现代企业信息化架构中,局域网电脑控制软件如同“指挥官”,通过自适应指令分发算法动态调整指令发送节奏与数据量,确保不同性能的终端设备高效运行。基于C++语言,利用套接字实现稳定连接和线程同步管理,结合实时状态反馈,优化指令分发策略,提升整体管控效率,保障网络稳定,助力数字化办公。
69 19
|
2月前
|
存储 算法 测试技术
【C++数据结构——树】二叉树的遍历算法(头歌教学实验平台习题) 【合集】
本任务旨在实现二叉树的遍历,包括先序、中序、后序和层次遍历。首先介绍了二叉树的基本概念与结构定义,并通过C++代码示例展示了如何定义二叉树节点及构建二叉树。接着详细讲解了四种遍历方法的递归实现逻辑,以及层次遍历中队列的应用。最后提供了测试用例和预期输出,确保代码正确性。通过这些内容,帮助读者理解并掌握二叉树遍历的核心思想与实现技巧。
55 2
|
3月前
|
存储 算法 安全
基于红黑树的局域网上网行为控制C++ 算法解析
在当今网络环境中,局域网上网行为控制对企业和学校至关重要。本文探讨了一种基于红黑树数据结构的高效算法,用于管理用户的上网行为,如IP地址、上网时长、访问网站类别和流量使用情况。通过红黑树的自平衡特性,确保了高效的查找、插入和删除操作。文中提供了C++代码示例,展示了如何实现该算法,并强调其在网络管理中的应用价值。
|
2月前
|
存储 算法 安全
基于哈希表的文件共享平台 C++ 算法实现与分析
在数字化时代,文件共享平台不可或缺。本文探讨哈希表在文件共享中的应用,包括原理、优势及C++实现。哈希表通过键值对快速访问文件元数据(如文件名、大小、位置等),查找时间复杂度为O(1),显著提升查找速度和用户体验。代码示例展示了文件上传和搜索功能,实际应用中需解决哈希冲突、动态扩容和线程安全等问题,以优化性能。
|
3月前
|
算法 安全 C++
用 C++ 算法控制员工上网的软件,关键逻辑是啥?来深度解读下
在企业信息化管理中,控制员工上网的软件成为保障网络秩序与提升办公效率的关键工具。该软件基于C++语言,融合红黑树、令牌桶和滑动窗口等算法,实现网址精准过滤、流量均衡分配及异常连接监测。通过高效的数据结构与算法设计,确保企业网络资源优化配置与安全防护升级,同时尊重员工权益,助力企业数字化发展。
68 4
|
17天前
|
编译器 C++ 开发者
【C++篇】深度解析类与对象(下)
在上一篇博客中,我们学习了C++的基础类与对象概念,包括类的定义、对象的使用和构造函数的作用。在这一篇,我们将深入探讨C++类的一些重要特性,如构造函数的高级用法、类型转换、static成员、友元、内部类、匿名对象,以及对象拷贝优化等。这些内容可以帮助你更好地理解和应用面向对象编程的核心理念,提升代码的健壮性、灵活性和可维护性。
|
20天前
|
编译器 C语言 C++
类和对象的简述(c++篇)
类和对象的简述(c++篇)