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
锻炼了堆的基本操作,比如插入和删除这种比较基础的操作
#include <iostream>
#define MaxN 10001
#define Min -10001
using namespace std;
typedef struct HeapNode* Heap;
struct HeapNode {
int size;
int data[MaxN];
HeapNode() :data{ Min },size(0) {}
};
/*插入最小堆*/
void insert(Heap H, int x) {
if (H->size>= MaxN) return;
else {
H->size++;
int i = H->size;
H->data[i] = x;
for (i = H->size; H->data[i / 2]>x; i = i / 2) {
H->data[i] = H->data[i / 2];
}
H->data[i] = x;
}
}
void Trace(Heap H, int num) {
if (num <= 0) return;
else {
int i = num;
for (; i >= 1; i = i / 2) {
if (i > 1) cout << H->data[i] << " ";
else if (i == 1) cout << H->data[i];
}
}
}
int main() {
/*向堆中插入元素*/
/*打印对应下标的根节点上的路径*/
int N, M, x,num;
cin >> N >> M; /*读入N,M*/
Heap H=new HeapNode; /*构建一个堆*/
for (int i = 0; i < N; i++) {
cin >> x;
insert(H, x);
}
for (int i = 0; i < M; i++) {
cin >> num;
Trace(H, num);
cout << endl;
}
return 0;
}