将一系列给定数字插入一个初始为空的小顶堆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
这题思路还是非常简单的,并且在mooc陈越老师的课上已经说过了,只要注意一般构建堆的时候回设置一个哨兵节点,在H[0]的位置,最小堆就放最小值,最大堆就放最大值,这样在之后的插入操作中就会简洁很多(和链表中设置头结点的意义差不多)。另外,由于设置了哨兵节点,输入的数据是从下标1开始的,那么根节点和左子树结点的下标就是2倍的关系,也就是leftroot=root/2。其他的就没什么了。
//05-树7 堆中的路径 (25 分)
#include<stdio.h>
#define Maxsize 1001
int H[Maxsize], size;void Initialize(void);
void Insert(int Data);
void search(int i);int main()
{
int N=0, M=0, i=0, Data=0, position=0;
scanf("%d%d", &N, &M);
Initialize(); //堆的初始化
for(i=0;i<N;i++){
scanf("%d", &Data);
Insert(Data); //堆的插入
}
for(i=0;i<M;i++){
scanf("%d", &position);
search(position);
if(i!=M-1) printf("\n");
}
return 0;
}void Initialize(void)
{
size=0;
H[0]=-10001;
}void Insert(int Data)
{
int i=0;
for(i=++size;H[i/2]>Data;i/=2){
H[i]=H[i/2];
}
H[i]=Data;
}void search(int i)
{
for(;i>0;i/=2){
printf("%d", H[i]);
if(i!=1) printf(" ");
}
}