堆的C++基本实现

原创于 2019-03-02 10:36:44 发布 · 95 阅读

CC 4.0 BY-SA版权

本文详细介绍了如何使用C++实现堆数据结构的基本操作，包括插入(push)和删除(pop)元素。通过具体代码示例，展示了堆的维护过程，如元素上浮和下沉的算法细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <algorithm>
#include <string.h>
#include <string>
#include <set>
#include <list>
#include <cstdio>
#include <math.h>
#include <vector>
using namespace std;

typedef long long ll;

const int MAX = 1001;

int heap[MAX], sz = 0;

void push(int x){
	
	int i = sz++;
	
	while(i > 0){
		
		int p = (i-1)/2;
		if(heap[p] <= x) break;
		
		heap[i] = heap[p];
		i = p;
	}
	
	heap[i] = x;
}

int pop(){
	
	int ret = heap[0];
	
	int x = heap[--sz];
	
	int i = 0;
	while(2 * i + 1 < sz){
		int a = 2 * i + 1, b = 2 * i + 2;
		if(b<sz && heap[a]>heap[b]) a = b;
		
		if(heap[a]>=x) break;
		
		heap[i] = heap[a];
		i = a;
	}
	
	heap[i] = x;
	
	return ret;
}

int main(){
	
	ios::sync_with_stdio(false);
	
	push(0);
	push(3);
	push(2);
	
	for(int i=0;i<3;i++){
		cout<<heap[i]<<" ";
	}
	cout<<endl;
	
	push(1);
	for(int i=0;i<4;i++){
		cout<<heap[i]<<" ";
	}
	cout<<endl;
	
	pop();
	for(int i=0;i<3;i++){
		cout<<heap[i]<<" ";
	}
	cout<<endl;
	
	return 0;
}