非优先队列的小顶堆

对应洛谷题目堆模板
小顶堆就是一直把最小的元素放根节点,父节点小于子节点,尾部输入,顶部输出;
如果i代表当前位置的下标,当2*i>len(堆对应数组的长度)代表这个节点没有子节点
对应两种操作,尾部插入元素和删除顶堆元素,当删除了顶堆元素时,把尾部元素赋值给顶部位置,然后删除尾部元素,然后再向下调整堆;
对应代码

#include<iostream>
using namespace std;
const int MAXN=1e6+5;
int  heap[MAXN]; 
int len=0;//记录数组长度,当前堆最大序号 
void push(int x)//进堆无返回值 
 {
 	heap[++len]=x;//从1开始
	 int sign=len;//实时记录要用的位置
	 while(sign>1&&heap[sign]<heap[sign/2])//当前点的父节点为sign/2,循环确定最小位置 
	 {
	 	swap(heap[sign],heap[sign/2]);
	 	sign=sign/2; 
	 } 

 }
void pop(int x)//删除表头,并调整堆,就是下沉
 {
 	heap[1]=heap[len--];//赋值后把尾也删除,减小堆的长度
	  int sign=1;//记录当前位置
	  while(2*sign<=len)//2*sign>len说明没有子节点,2*sign+1>len说明没有右节点 
	  {
	  	
	  	int son=2*sign;
	    if(son<len&&heap[son]>heap[son+1])//son<len先确定有右孩子然后比较孩子带大小,选择小的那个 
	    {
	    	son++;//右小 
		}
		if(heap[sign]>heap[son])
		{
				swap(heap[sign],heap[son]) ;
		         sign=son ;//改变当前位置; 
			 }	 
		else
		break;//不满足条件说明调整好了,结束 
	   } 
 } 
int main()
{
	int N;
	int xx;
	int signs;
	cin>>N;
	while(N--)
	{
		cin>>signs;
		if(signs==1)
		{
			cin>>xx;
			push(xx);
		}
		else if(signs==2)
		{
			cout<<heap[1]<<endl;;//小顶堆,所以最小的是第一个
			 
		}
		else
		pop(xx); 
	 } 
	return 0;
}

用因为不是优先队列scanf printf是因为用cin cout会差几秒tle
想要更详细的理解参考:stl

猜你喜欢

转载自blog.csdn.net/qq_45337491/article/details/106137029