05-树7 堆中的路径 (25分)
将一系列给定数字插入一个初始为空的小顶堆H[]。随后对任意给定的下标i,打印从H[i]到根结点的路径。
输入格式:
每组测试第1行包含2个正整数N和M(≤1000),分别是插入元素的个数、以及需要打印的路径条数。下一行给出区间[-10000, 10000]内的N个要被插入一个初始为空的小顶堆的整数。最后一行给出M个下标。
输出格式:
对输入中给出的每个下标i,在一行中输出从H[i]到根结点的路径上的数据。数字间以1个空格分隔,行末不得有多余空格。
输入样例:
5 3
46 23 26 24 10
5 4 3
输出样例:
24 23 10
46 23 10
26 10
作者
陈越
单位
浙江大学
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
#include<iostream>
using namespace std;
#define MINDATA -10001
struct HNode{
int *data;
int size;
int capacity;
};
typedef HNode *maxHeap;
typedef HNode *minHeap;
minHeap CreateHeap(int maxSize){
minHeap h = (minHeap) malloc(sizeof(struct HNode));
h->data = (int *)malloc((maxSize + 1) * sizeof(int));
h->size = 0;
h->capacity = maxSize;
h->data[0] = MINDATA;
return h;
}
bool IsFull(minHeap h){
return(h->size == h->capacity);
}
bool Insert(minHeap h, int x){
int i;
if(IsFull(h)){
return false;
}
i = ++h->size;
for(; h->data[i/2] > x; i /= 2){
h->data[i] = h->data[i/2];
}
h->data[i] = x;
return true;
}
int main(){
int n1, n2;
int data;
int num;
scanf("%d%d", &n1, &n2);
minHeap h = CreateHeap(n1);
for(int i = 0; i < n1; i++){
scanf("%d", &data);
Insert(h, data);
}
for(int i = 0; i < n2; i++){
scanf("%d", &num);
int j = num;
// printf("%d", h->data[num]);
while(j != 1){
printf("%d ", h->data[j]);
j = j / 2;
}
printf("%d\n", h->data[j]);
}
return 0;
}