堆操作,版本二
這個和上個版本類似,略有不同
//二叉堆(數(shù)組存儲)的插入,刪除堆頂元素(以小根堆為例,即堆頂元素最小) #include<cstdio> #include<iostream> #include<algorithm> using namespace std; const int Max_size = (int)1e3; //表示堆中能夠容納元素個數(shù)的最大值 int size = 0,Heap[Max_size]; //記錄當(dāng)前堆中元素下標(biāo)最大值//往堆中插入元素,數(shù)組下標(biāo)從0開始 void push(int value){ //value表示插入堆的值Heap[size] = value; //一開始元素接在堆的最后面int current = size; //表示當(dāng)前遍歷到的節(jié)點(diǎn)int father = (current-1) / 2; //表示當(dāng)前遍歷到的堆中元素的父親節(jié)點(diǎn)的下標(biāo)while(current > 0 && Heap[current] < Heap[father]){ //表示此時節(jié)點(diǎn)"上溢",當(dāng)前節(jié)點(diǎn)與其父節(jié)點(diǎn)對換.swap(Heap[current],Heap[father]);current = father;father = (current-1) / 2; //繼續(xù)向上遍歷}++size; }/**刪除棧頂元素*刪除棧頂元素時,不可直接將size的值減一后就結(jié)束,這樣的話會破壞整個*二叉堆的結(jié)構(gòu),因此我們在刪除堆頂元素之后還要調(diào)整堆的結(jié)構(gòu),使其成為新的堆*并且有新的堆頂元素*/ int pop(){int topHeap = Heap[--size]; Heap[0] = topHeap; //首先將最后一個元素替換堆頂元素,這樣的話堆頂元素則被刪除,然后再是size減一表示堆中元素數(shù)目減一int current = 0; //當(dāng)前遍歷到的節(jié)點(diǎn)的下標(biāo)int lchild = current*2 + 1; //當(dāng)前遍歷到的節(jié)點(diǎn)的左孩子和右孩子節(jié)點(diǎn)的下標(biāo)int rchild = current*2 + 2;while(lchild < size && min(Heap[lchild],Heap[rchild]) < Heap[current]){if(Heap[lchild] < Heap[rchild]){swap(Heap[lchild],Heap[current]);current = lchild; //"下溢"的過程}else{swap(Heap[rchild],Heap[current]);current = rchild;}lchild = current*2 + 1;rchild = current*2 + 2;}return topHeap; }//獲取堆頂元素 int top(){return Heap[0]; }//輸出堆中元素 void output(){for(int i = 0;i < size;++i){printf("%d ",Heap[i]);}printf("\n"); } int main(){int a[6];cout<<"請輸入堆中元素:"; for(int i = 0;i < 6;++i){cin>>a[i];push(a[i]);}printf("小根二叉堆堆頂元素為:%d\n",top());cout<<"堆中所有元素為:"<<endl; output();return 0; }?
總結(jié)
- 上一篇: 堆的基本操作
- 下一篇: Hadoop 详细配置文档