pta ds 7-1 堆中的路径 (25分) stl heap
7-1 堆中的路径 (25分)
将一系列给定数字插入一个初始为空的小顶堆H[]
。随后对任意给定的下标i
,打印从H[i]
到根结点的路径。
输入格式:
每组测试第1行包含2个正整数N和M(≤),分别是插入元素的个数、以及需要打印的路径条数。下一行给出区间[-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
作者
陈越
单位
浙江大学
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
我使用的是stl中的heap 学习堆过程不推荐使用
stl的heap使用方法可以参考https://blog.csdn.net/zsc2014030403015/article/details/45872737
#include<stdio.h> #include<stdlib.h> #include<iostream> #include<algorithm> using namespace std; bool cmp(int a,int b){ return a>b; } int main(){ int n,m; cin>>n>>m; int a[n+1]; for(int i=1;i<=n;i++){ cin>>a[i]; push_heap(a+1,a+i+1,cmp); } for(int i=0;i<m;i++){ int temp; cin>>temp; while(1){ cout<<a[temp]; temp/=2; if(temp!=0)cout<<" "; else break; } cout<<endl; } return 0; }