7-5 堆中的路径 (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
代码实现(C语言)
#include<stdio.h>
#define MaxSize 1000
int main()
{
int heap[MaxSize]={-10000}; /* 数组初始化 */
int n_data, n_test, temp, last=1; /* last为堆size */
scanf("%d %d", &n_data, &n_test); /* 读入第一行堆的元素数已经测试组数 */
for(int i = 0; i < n_data; i++)
{
scanf("%d", &temp);
/* 插入法建堆 */
if(last == 1)
{
heap[last] = temp;
}
else
{
int j;
/* 上滤的核心代码 */
for(j=last; heap[j/2]>temp&&j; j/=2)
heap[j] = heap[j/2];
heap[j] = temp;
}
last++;
}
/* 输出部分 */
for(int i = 0; i < n_test; i++)
{
scanf("%d", &temp);
printf("%d", heap[temp]);
temp /= 2;
while(temp)
{
printf(" %d", heap[temp]);
temp /= 2;
}
printf("\n");
}
return 0;
}