PTA7-9 堆中的路径 (C++)

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

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

所谓的小根堆其实就跟是一颗二叉树，保证每一层的任何一个数字都小于下一层的任何一个数字

#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define int long long
#define PII pair<int,int>
#define fi first
#define pb push_back
#define sec second
#define endl '\n'
#define INF 0x3f3f3f3f
const int N=2e5+10;
const int M=1e6+10;
int n,m,k,T;
int s[N];
signed main(){
	cin >> n >> m;
	for(int i=1;i<=n;i++){
		cin >> s[i];
		int tmp = i ;
		//取当前下标 
		while(tmp!=1){
			if(s[tmp/2]>s[tmp]){
				swap(s[tmp/2],s[tmp]);
			}
			//找父亲节点是否比当前节点大，如果大的话要交换 
			tmp/=2;
		}
	}
	int a ;
	for(int i=1;i<=m;i++){
		cin >> a ;
		int tmp = a ;
		while(tmp!=1){
			cout << s[tmp] << ' ' ; 
			tmp/=2;
		}
		cout << s[1];
		cout << endl;
		//打印路径 
	}
	return 0;
}