原题链接如下:
AcWing 838. 堆排序
对堆的具体介绍,本人水平和时间都有限,就不在这里赘述了,主要是把从y总那里学来的优雅的手写堆排序的代码复现一下,以加深自己的理解
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 100010;
int n,m;
int h[N],cnt;
void down(int u)
{
//t我们可以看成指针,每次指向继续要向下down操作的位置
int t = u;
if(2*u<=cnt&&h[2*u]<h[t])t = 2*u;
if(2*u+1<=cnt&&h[2*u+1]<h[t])t = 2*u+1;
if(u!=t)
{
swap(h[t],h[u]);
down(t);
}
}
int main()
{
cin>>n>>m;
for(int i = 1;i<=n;i++)scanf("%d",&h[i]);
cnt = n;
//创建堆
for(int i = n/2;i;i--)down(i);
//每次都取出堆中的首元素
while(m--)
{
printf("%d ",h[1]);
//堆首元素被取出来之后,尾部元素顶上,把尾部元素向下沉,找到它合适的位置
h[1] = h[cnt--];
down(1);
}
return 0;
}
这里补充解释一点点我的里面t
的理解,一开始我并没有特别想明白t
的作用(特别是y总的down操作里面还有递归,让我这个菜鸡就更加迷糊了),后面我自己仔细模拟了一下down
操作的过程,发现我们可以把t
看成一个指针,每次进行down操作的时候,我们先让t
指向传入参数的下标,然后进行比较,由于down
操作的作用是把大的元素往堆的下面沉,所以我们每次递归传入的参数,肯定都是自己、左子节点、右子节点三个中最大的那个,所以我们就用t来维护每次要递归往下传的参数,简单来说,t每次存储的就是三个节点中最大的那个的位置,注意是位置!!(在堆中的位置)