标签:sz int 838 堆排序 down -- 小根堆 节点 AcWing
点击查看代码
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int h[N], sz;
void down(int u)
{
int t = u;
if (2 * u <= sz && h[2 * u] < h[t]) t = 2 * u;
if (2 * u + 1 <= sz && h[2 * u + 1] < h[t]) t = 2 * u + 1;
if (u != t) {
swap(h[t], h[u]);
down(t);
}
}
int main()
{
int n, m;
cin >> n >> m;
for (int i = 1; i <= n; i ++) cin >> h[i];
sz = n;
for (int i = sz / 2; i; i --) down(i);
while (m --) {
cout << h[1] << ' ';
h[1] = h[sz];
sz --;
down(1);
}
return 0;
}
- 维护一个小根堆(二叉树,且每个父节点都比两个孩子节点要小),小根堆的一个重要操作是 down,作用是把当前节点 u 和两个孩子节点的值进行比较,将较小的节点 t 放到父节点的位置,然后递归地去处理 u 被交换后,目前所在的这颗子树;(u 现在的位置是 t)
- 建立小根堆的操作:从最下面的第二层开始,对每一层的节点执行 down,也就是数组下标 i 从 sz / 2 到 1,down( i ),循环结束后即为小根堆;
- 删除操作:用最后一个节点的值替换掉根节点的值,然后对根节点执行 down( 1 ),总节点数 sz - 1;
标签:sz,int,838,堆排序,down,--,小根堆,节点,AcWing 来源: https://www.cnblogs.com/wKingYu/p/16218545.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。