天天看点

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

###思路:

本题以数组存储最小堆。依次插入,核心代码在 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]);
}