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

---

解答

知识点:
利用堆的特性 数组建立最大最小堆
思路:
利用数组存最小堆
数组[0]存哨兵

#include<iostream>
using namespace std;
int min_heap[100001];
int size=0;
int main()
{
    min_heap[0]=-1000000;//哨兵保证最小
    int N,M;
    int elem;
    int i,j;
    cin>>N>>M;
    for(i=0;i<N;i++){
        cin>>elem;
        for(j=++size;elem<min_heap[j/2];j=j/2) min_heap[j]=min_heap[j/2];
        min_heap[j]=elem;
    }
    int index;
    for(int i=0;i<M;i++){
        cin>>index;
        cout<<min_heap[index];
        for(int j=index/2;j>=1;j=j/2){
            cout<<" "<<min_heap[j];
        }
        cout<<endl;
    }
    
    return 0;
}