PTA 05-树7 堆中的路径

PTA 05-树7 堆中的路径

题目链接

将一系列给定数字插入一个初始为空的小顶堆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
结尾无空行

Code

#include<iostream>
#define MaxN 1001
#define Min -10001
using namespace std;
int heap[MaxN];
int size;
//堆初始化
void Create(){
    size=0;
    heap[0]=Min;
}
//将num值插入到heap中
void Insert(int num){
    int i=++size;
    for(;heap[i/2]>num;i/=2){//注意这里是和Num作比较
        heap[i]=heap[i/2];
    }//逐步向下移动位置，最后适合插入的位置为i
    heap[i]=num;
}
//从某个位置开始打印至树根
void print(int pos){
    printf("%d",heap[pos]);
    int parent=pos/2;
    while(parent>0){
        printf(" %d",heap[parent]);
        parent/=2;
    }
}
int main(){
    int N,M;
    scanf("%d %d\n",&N,&M);
    Create();
    for(int i=0;i<N;i++){
        int num;
        scanf("%d",&num);
        Insert(num);//插入堆中
    }
    for(int i=0;i<M;i++){
        int pos;
        scanf("%d",&pos);
        if(i!=0){
            printf("\n");
        }
       print(pos);
    }
    return 0;
}