堆的操作,用数组实现
包括插入操作--自底向上调整、删除操作--把最后一位赋值给堆顶,然后从堆顶向下调整
#include<iostream> #include<cstring> #include<cmath> #include<algorithm> #include<stack> #include<cstdio> #include<queue> using namespace std; //堆 //一种数组对象,可以被视为一棵完全二叉树,树中的每个节点都与数组中存放该节点中值得那个元素对应 int heap_size=0; int heap[10001]; //插入操作:自底向上 void put(int d){//首先看插入操作put int now,next; heap[++heap_size]=d;//首先把元素插入末尾 now=heap_size; while(now>1){ next=now>>1;//即next=now/2 if(heap[now]>=heap[next]) break;//这是小根堆,因为小的在上面 else swap(heap[next],heap[now]);//否则就交换,并且继续迭代 now=next; } } //删除操作: 取出堆顶元素,然后把最后一个赋值给堆顶,然后向下调整 int get(){//从堆中取出并删除一个元素get, int now,next,res; res=heap[1]; heap[1]=heap[heap_size--]; now=1; while(now*2<=heap_size){ next=now*2; if(next<heap_size&&heap[next]<heap[next+1]) next++; //用最小的与之交换 if(heap[now]<=heap[next]) break; swap(heap[now],heap[next]); now=next; } return res; } int main(){ int a[20]={0,3,5,1,7,6,4,2,5,4,1}; for(int i=1;i<=10;i++) put(a[i]); cout<<"堆中元素如下"<<endl; for(int i=1;i<=10;i++) cout<<heap[i]<<" "; cout<<endl; cout<<"最小的元素 "<<endl; cout<<get(); return 0; }
完整的:大根堆,可以实现排序
建堆的时候倒着建,for(int i=n/2;i>=1;i--) 这样可以保证每个子树的根都是最大的
排序的时候每次都把堆顶移到最后,然后n=i这样减少 for(int i=n;i>1;i--) swap(a[i],a[1]) downjust(1,i-1)
//堆: //a heap is a specialized tree-based data structure(树)、并查集、堆的实质其实都可以是树 //一般的快速数据结构:优先队列: //优先队列默认:大顶堆 (不用自己写函数) priority_queue<int> q1; //小顶堆 priority_queue<int,vector<int>,greater<int> > q2; //跟一般的数据结构(需要自己写函数的那种) //需要自己写函数 int heap[maxn]; //just like堆 //但是掌握堆的自顶向上和自底向上的两种写法是很重要的 //建堆的过程:自上向下调整,但是需要倒着枚举调整,调整所有的非叶子节点,从n/2到1,倒着调整,但是调整是自顶向下的 //包括删除节点的时候:删除堆顶元素,用最后一个元素覆盖第一个,然后调整downjust(1,n) void downadjust(int low,int high){ //向下调整:删除、建堆、堆排序 int i=low,j=2*i; while(j<=high){ //如果有孩子存在,且有孩子的值大于左孩子的值,就像后移,然后再跟i比较大小 if(j+1<=high&&heap[j+1]>heap[j]) j++; if(heap[j]>heap[i]){ //如果孩子比爸爸大,就要交换 swap(heap[j],heap[i]); i=j; j=2*i; } else break; //不然就退出 } } void create(){ for(int i=1;i<=n;i++) cin>>heap[i]; for(int i=n/2;i>=1;i--) downadjust(i,n); } //倒着调整:这样才能满足每个节点都是以他为父节点的最大值 //删除堆顶元素 void del(){ heap[1]=heap[n--]; downadjust(1,n); //调整堆顶 } //添加一个元素的时候:添加再最后面,然后自底向上调整 void upadjust(int high,int low){ int i=high,j=high/2; while(j>=low){ //总是和父亲比较的话,就不存在j++或者j--了,直接比较就可以了 if(heap[j]<heap[i]){ swap(heap[j],heap[i]); i=j; //如果父亲比小的话,就交换 j/=2; } else break; } } void insert(int x){ heap[++n]=x; //添加再末尾 upadjust(n,1); } //删除堆顶,添加末尾 //堆排序 //由于堆中堆顶元素是确定的最大的,所以每次都去堆顶元素和最末尾的元素交换,然后从堆顶进行自顶向下的调整 //这样每次都取出了最大堆顶元素,而且对的范围也在缩小 void heapsort(){ create(); //先建堆 for(int i=n;i>1;i--) { swap(a[i],a[1]); //交换 downadjust(1,i-1); //范围再缩小 } for(int i=1;i<=n;i++) cout<<heap[i]<<" "; //自小到大的序列 }