将一系列给定数字插入一个初始为空的小顶堆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
###思路:
本题以数组存储最小堆。依次插入,核心代码在 resort 函数。每次插入一个元素,放在数组最后,即树层次遍历的最后一个位置,然后和他的父亲节点做比较,如果它大于父亲节点的值,则满足最小堆的定义,否则将它和父亲节点做交换,循环直至到根节点或者小于父亲节点值。
最小堆:根节点的值比左右节点的值都小的二叉树,也叫小顶堆/小根堆。
最大堆:根节点的值比左右节点的值都大的二叉树,也叫大顶堆/大根堆。
#include<stdio.h>
#define N 1002
int h[N]={0};
void resort(int i);
void way2root(int i);
int main(){
int n, m;
scanf("%d%d",&n, &m);
for(int i = 1; i <= n; i++){//二叉树从1开始编码
scanf("%d",h+i);
resort(i);
}
while(m--){
int index;
scanf("%d",&index);
way2root(index);
}
return 0;
}
void resort(int i){
//把新插入的元素,放入到数组最后一个元素
//然后重新组织成最小堆
while(i/2&& h[i/2] > h[i]){
int tmp = h[i];
h[i] = h[i/2];
h[i/2] = tmp;
i /= 2;
}
}
void way2root(int i){
while(i != 1){
printf("%d ",h[i]);
i /= 2;
}
printf("%d\n",h[1]);
}