PTA 7-5 堆中的路径（25 分）STL方法，简化版

7-5 堆中的路径（25 分）

将一系列给定数字插入一个初始为空的小顶堆H[]。随后对任意给定的下标i，打印从H[i]到根结点的路径。

输入格式:

每组测试第1行包含2个正整数$N$和$M$($\le 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

上一次遇到这个问题，就想到调用stl的make_heap()函数构造堆来实现功能，make_heap()函数构造的是大顶堆，题目中要求构造小顶堆，而且边插入边构造堆，这点需要注意一下，make_heap()函数构造小顶堆，参考优先队列改成小顶堆的方式，利用greater<int>()，构造小顶堆

#include<bits/stdc++.h>  
using namespace std;    
    
int main()      
{         
    int n,m,i,test,count,t;int b[1005];
    vector<int> a;
    scanf("%d%d",&n,&m); 
    for(i=1;i<=n;i++)   
    {  
        scanf("%d",&t);  
        a.push_back(t);
	    make_heap(a.begin(),a.end(),greater<int>());//构造小顶堆的方法 
    } 
	for(i=0;i<n;i++)
		b[i+1]=a[i];
    for(i=1;i<=m;i++)  
    {  
        scanf("%d",&test);  
        count=0;  
        while(test)  
        {  
	        count++;  
	        if(count==1)  
	            printf("%d",b[test]);  
	        else  
	            printf(" %d",b[test]);  
	        test/=2;  
        }  
        if(i!=m)  
            printf("\n");  
    }  
    return 0 ;      
}

高效简洁