首页 > 编程语言 >c++算法学习笔记 (19) 堆

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

时间:2024-04-06 19:58:17浏览次数:34  
标签:sz 19 up c++ down int 算法 kk heap

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,输出当前集合中的最小值;
  3. DM,删除当前集合中的最小值(数据保证此时的最小值唯一);
  4. D k,删除第 k 个插入的数;
  5. C k x,修改第 k 个插入的数,将其变为 x;

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

输入格式

第一行包含整数 N。

接下来 N 行,每行包含一个操作指令,操作指令为 I xPMDMD k 或 C 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;
}

 

标签:sz,19,up,c++,down,int,算法,kk,heap
From: https://blog.csdn.net/l141930402/article/details/137374881

相关文章

  • C/C++预处理过程
    目录前言:1.预定义符号2. #define定义常量3. #define定义宏4.带有副作用的宏参数5.  宏替换的规则6. 宏和函数的对比7.#和##8. 命名约定9. #undef10. 命令行定义11. 条件编译12. 头文件的包含13. 其他预处理指令总结:。前言:本篇介绍c/c+......
  • C++模版简单认识与使用
    目录前言:1.泛型编程2.函数模版3.类模版为什么要有类模版?使用typedef不行吗?类模版只能显示实例化:注意类名与类型的区别:注意类模版最好不要声明和定义分离:总结:前言:正如标题而言,这里只是对模版的简单认识与使用,方便后面博客介绍stl中一些容器的实现,更复杂详细的模版......
  • Python常用算法--排序算法【附源码】
    应用具体python案例方式展示各种排序的要点,特别是希尔排序、插入排序、选择排序、冒泡排序、堆排序、快速排序、归并排序七个具体的排序算法。一、希尔排序:解释:希尔排序(ShellSort)是一种插入排序的改进版本,也被称为缩小增量排序。希尔排序通过比较相距一定间隔的元素,将大间隔......
  • FJSP:蜣螂优化算法( Dung beetle optimizer, DBO)求解柔性作业车间调度问题(FJSP),提供MAT
    一、柔性作业车间调度问题柔性作业车间调度问题(FlexibleJobShopSchedulingProblem,FJSP),是一种经典的组合优化问题。在FJSP问题中,有多个作业需要在多个机器上进行加工,每个作业由一系列工序组成,每个工序需要在特定的机器上完成。同时,每个机器一次只能处理一个工序,且每个工......
  • FJSP:霸王龙优化算法(Tyrannosaurus optimization,TROA)求解柔性作业车间调度问题(FJSP),提供
    一、柔性作业车间调度问题柔性作业车间调度问题(FlexibleJobShopSchedulingProblem,FJSP),是一种经典的组合优化问题。在FJSP问题中,有多个作业需要在多个机器上进行加工,每个作业由一系列工序组成,每个工序需要在特定的机器上完成。同时,每个机器一次只能处理一个工序,且每个工......
  • 异或运算在算法中的神奇应用
    1.什么是异或两个二进制数进行异或运算时,每一位上的数相同则结果为0,不同则结果为1。示例:6^7=?转化成二进制:6=1107=1116^7=110^111=001=1简单记:异或就是二进制的无进位相加。还有个同或运算:相同为1,不同为0,和异或是反的。2.异或运算的特性任何数与0异或,结果还是这......
  • CF1929B Sasha and the Drawing 题解
    CF1929B题意给定一个\(n\timesn\)的正方形,已知正方形最多有\(4\timesn-2\)条对角线,要求要有至少\(k\)条对角线经过至少一块黑色方格,求至少要将几条对角线涂成黑色。分析分类讨论:当\(k<=4\timesn-4\)时,就只需要在上下两侧图就行,所以答案是\([\frac{k}{2}]\)。当......
  • CF1915B Not Quite Latin Square 题解
    CF1915B题意给出一个\(3\)行\(3\)列的字符矩形,其中每行都有字符ABC各一个组成,现有一个字符未知,求出未知字符。思路就是说每个字符都应该出现\(3\)次,所以我们只要找到出现两次的字符即可。ACCode#include<bits/stdc++.h>usingnamespacestd;intt;chara[10][10......
  • P6680 [CCO2019] Marshmallow Molecules 题解
    P6680题意一个\(n\)点\(m\)边的图,图无重边,无自环。满足这样一条性质:如果三边互不相等,则三边可以构成三角形。思路思路简单,用集合的思想来做。引用一下K0stlin大佬的性质:题目中的操作等价于将一个点大于某个儿子的儿子们赋给这个儿子(这里的儿子表示这个点有出边连向的......
  • QT和C++排列组合
    界面比较简洁,如有需要请大家自行完善!!!头文件#pragmaonce#include<QtWidgets/QMainWindow>#include"ui_text.h"classtext:publicQMainWindow{  Q_OBJECTpublic:  text(QWidget*parent=nullptr);  ~text();  voidParseStringToVector(con......