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
自顶向下和自底向上都是沿着根到叶的路径调整,时间复杂度为O(logN)即树高。
插入操作:新元素插入到堆的末尾,沿着到根的路径自底向上调整;
删除操作:删除堆根元素,将堆末尾元素移到根,选择正确的道路自顶向下调整
#include <stdio.h>
#include <malloc.h>
#define max 1000
#define maxData 10001
#define minData -10001
typedef int ElementType;
typedef struct HNode{
ElementType* Data;
int size;
int MaxSize;
}*Heap;
typedef Heap MaxHeap;
typedef Heap MinHeap;
MinHeap creatNullMinHeap(int size){
MinHeap H = (MinHeap)malloc(sizeof(struct HNode));
H->Data = (ElementType*)malloc(sizeof(ElementType)*(size+1));
H->size = 0;
H->MaxSize = size;
H->Data[0] = minData; //设置一个哨兵,根为1
return H;
}
int IsFull(MinHeap H){
if(H->size==H->MaxSize) return 1;
else return 0;
}
int IsEmpty(MinHeap H){
if(H->size==0) return 1;
else return 0;
}
void AdjustUp(MinHeap H,int position){
//沿着position到根的路径自底向上调整
int child = position;
H->Data[0] = H->Data[child];
for(int parent = child/2;H->Data[parent]>H->Data[0];parent/=2){ //循环边界:双亲小于孩子及堆的上边界
H->Data[child] = H->Data[parent]; //双亲下移
child = parent;
}
H->Data[child] = H->Data[0];
}
void AdjustDown(MinHeap H,int position){
//position的左右子树都是小(大)根堆,自顶向下调整
int parent = position;
H->Data[0] = H->Data[parent];
for(int child = parent*2;child<=H->size;child*=2){//与自底向上不同 ,循环边界为堆的下边界
if(child!=H->size&&H->Data[child+1]<H->Data[child])
child++;
if(H->Data[child]<H->Data[0]){ //孩子小于双亲
H->Data[parent] = H->Data[child]; //孩子上移
parent = child;
}
else //与自底向上调整不同,若找到了正确的位置应跳出循环
break;
}
H->Data[parent] = H->Data[0];
}
void InsertMinHeap(MinHeap H,int e){
if(IsFull(H)) return;
H->Data[++H->size] = e;
AdjustUp(H,H->size);
}
ElementType DeleteMinHeap(MinHeap H){
if(IsEmpty(H)){
//printf("堆空\n");
return -1;
}
ElementType x = H->Data[1];
H->Data[1] = H->Data[H->size--];
AdjustDown(H,1);
return x;
}
void BuildMinHeap(MinHeap H){
for(int i=H->size/2;i>0;i--){
AdjustDown(H,i);
}
}
void PrintLoad(MinHeap H,int position){
printf("%d",H->Data[position]);
for(int i=position/2;i>0;i/=2){
printf(" %d",H->Data[i]);
}
printf("\n");
}
int main(){
//freopen("1.txt","r",stdin);
int N,M,position;
scanf("%d %d\n",&N,&M);
MinHeap H = creatNullMinHeap(max);
// for(int i=1;i<=N;i++){
// scanf("%d",&H->Data[i]);
// H->size++;
// }
// BuildMinHeap(H);
ElementType x;
while(N--){
scanf("%d",&x);
InsertMinHeap(H,x);
}
while(M--){
scanf("%d",&position);
PrintLoad(H,position);
}
return 0;
}