编写代码,实现最小堆(Min-Heap)的操作。
输入格式:
第一行是两个不大于1000
的正整数N
和K
,用空格间隔。其中N
是堆的容量,需创建一个容量为N
的堆。
接下来K
行,是对这个堆的依次的K
项插入或删除操作:用 1 x
表示插入元素x;用 -1
表示删除堆顶。
接下来一行是一个不大于1000
的正整数M
,
接下来一行是M
个整数(在整型范围内),用空格间隔,
要求将这M
个整数组成的列表调整为一个最小堆。
输出格式:
对于第一个堆的K
项操作,每次操作后,在一行中依次序打印堆元素,元素间使用1个空格分隔;
对于第二个堆,在调整完成后,在一行中依次序打印堆元素,元素间使用1个空格分隔。
输入样例:
10 8
1 1
1 2
1 3
1 4
1 5
-1
1 6
-1
8
1 2 3 4 5 6 7 8
输出样例:
1
1 2
1 2 3
1 2 3 4
1 2 3 4 5
2 4 3 5
2 4 3 5 6
3 4 6 5
1 2 3 4 5 6 7 8
#include<iostream>
using namespace std;
struct heap{
int data[1001];
int size;
int capacity;
heap(int cap){
size = 0;
capacity = cap;
}
};
typedef struct heap* Heap;
void percolateUp(Heap h, int index){
int i, tmp = h->data[index];
for(i = index; i / 2 > 0 && tmp < h->data[i / 2]; i /= 2)
h->data[i] = h->data[i / 2];
h->data[i] = tmp;
}
void percolateDown(Heap h, int index){
int i, tmp = h->data[index], child;
for(i = index; i * 2 <= h->size; i = child){
child = i * 2;
if(child < h->size && h->data[child + 1] < h->data[child]) child++;
if(tmp > h->data[child]) h->data[i] = h->data[child];
else break;
}
h->data[i] = tmp;
}
void insert(Heap h, int key){
if(h->size == h->capacity) return ;
h->data[++h->size] = key;
percolateUp(h, h->size);
}
void remove(Heap h){
if(h->size == 0) return ;
h->data[1] = h->data[h->size];
h->size--;
percolateDown(h, 1);
}
void print(Heap h){
for(int i = 1; i < h->size; i++)
cout << h->data[i] << ' ';
cout << h->data[h->size] << endl;
}
int main(){
int N, K, M;
int tmp;
cin >> N >> K;
Heap h = new heap(N);
for(int i = 0; i < K; i++){
cin >> tmp;
if(tmp == 1){
cin >> tmp;
insert(h, tmp);
}else{
remove(h);
}
print(h);
}
cin >> M;
Heap h2 = new heap(1001);
for(int i = 0; i < M; i++){
cin >> tmp;
h2->data[++h2->size] = tmp;
}
for(int i = h2->size / 2; i > 0; i--){
percolateDown(h2, i);
}
print(h2);
return 0;
}