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;
}


相关文章
|
4天前
|
算法 C++
算法笔记:递归(c++实现)
算法笔记:递归(c++实现)
|
19小时前
|
算法 前端开发 Linux
【常用技巧】C++ STL容器操作:6种常用场景算法
STL在Linux C++中使用的非常普遍,掌握并合适的使用各种容器至关重要!
26 10
|
2天前
|
存储 算法 C++
【数据结构与算法】:带你手搓顺序表(C/C++篇)
【数据结构与算法】:带你手搓顺序表(C/C++篇)
|
2天前
|
算法 C++
【数据结构与算法】:关于时间复杂度与空间复杂度的计算(C/C++篇)——含Leetcode刷题-2
【数据结构与算法】:关于时间复杂度与空间复杂度的计算(C/C++篇)——含Leetcode刷题
|
2天前
|
算法 C++
【数据结构与算法】:关于时间复杂度与空间复杂度的计算(C/C++篇)——含Leetcode刷题-1
【数据结构与算法】:关于时间复杂度与空间复杂度的计算(C/C++篇)——含Leetcode刷题
|
4天前
|
算法 C++
【c/c++算法】曼哈顿算法简单运用
【c/c++算法】曼哈顿算法简单运用
|
2天前
|
算法 JavaScript 决策智能
基于禁忌搜索算法的TSP路径规划matlab仿真
**摘要:** 使用禁忌搜索算法解决旅行商问题(TSP),在MATLAB2022a中实现路径规划,显示优化曲线与路线图。TSP寻找最短城市访问路径,算法通过避免局部最优,利用禁忌列表不断调整顺序。关键步骤包括初始路径选择、邻域搜索、解评估、选择及禁忌列表更新。过程示意图展示搜索效果。
|
2天前
|
机器学习/深度学习 算法
基于BP神经网络和小波变换特征提取的烟草香型分类算法matlab仿真,分为浓香型,清香型和中间香型
```markdown 探索烟草香型分类:使用Matlab2022a中的BP神经网络结合小波变换。小波分析揭示香气成分的局部特征,降低维度,PCA等用于特征选择。BP网络随后处理这些特征,以区分浓香、清香和中间香型。 ```
|
3天前
|
机器学习/深度学习 算法
m基于PSO-GRU粒子群优化长门控循环单元网络的电力负荷数据预测算法matlab仿真
摘要: 在MATLAB 2022a中,对比了电力负荷预测算法优化前后的效果。优化前为&quot;Ttttttt111222&quot;,优化后为&quot;Tttttttt333444&quot;,明显改进体现为&quot;Tttttttttt5555&quot;。该算法结合了粒子群优化(PSO)和长门控循环单元(GRU)网络,利用PSO优化GRU的超参数,提升预测准确性和稳定性。PSO模仿鸟群行为寻找最优解,而GRU通过更新门和重置门处理长期依赖问题。核心MATLAB程序展示了训练和预测过程,包括使用&#39;adam&#39;优化器和超参数调整,最终评估并保存预测结果。
9 0
|
4天前
|
算法 安全
基于龙格库塔算法的SIR病毒扩散预测matlab仿真
该程序使用龙格库塔算法实现SIR模型预测病毒扩散,输出易感、感染和康复人群曲线。在MATLAB2022a中运行显示预测结果。核心代码设置时间区间、参数,并定义微分方程组,通过Runge-Kutta方法求解。SIR模型描述三类人群动态变化,常微分方程组刻画相互转化。模型用于预测疫情趋势,支持公共卫生决策,但也存在局限性,如忽略空间结构和人口异质性。

热门文章

最新文章