7-1 堆中的路径(25 分)

将一系列给定数字插入一个初始为空的小顶堆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
/*
    
      在这题中,很多同学可能会有疑惑,为什么会是这个输出结果?现在,来谈谈堆的性质,
    其必须满足儿子的值一定不小于父亲的值。除此之外,树的节点是按从上到下,从左到右的顺
    序紧凑排列的。
    
    第一轮:46
    heap[0] = 46;
    
    第二轮:23
    heap[0] = 23 , heap[1] = 46;
    
    第三轮:26
    heap[0] = 23 , heap[1] = 46 , heap[2] = 26;

    第四轮:24
    heap[0] = 23 , heap[1] = 24 , heap[2] = 26 , heap[3]=46;

    第五轮:10
    heap[0] = 10 , heap[1] = 23 , heap[2] = 26 , heap[3] = 46 , heap[4] = 24

*/

#include <iostream>
#include <algorithm>
#include <queue>
#include <string.h>
using namespace std;

const int MAX_N = 1001;

int heap[MAX_N],sz=0;

void push(int x){
	int i = sz++;
	while(i>0){
		int p = (i-1)/2;
		if(heap[p]<=x)	break;
		heap[i] = heap[p];
		i = p;
	}
	heap[i] = x;
}

bool flag;

void get_path(int x){
	for(int i=x;i>0;i=i/2){
		if(flag)
		cout<<" ";
		cout<<heap[i-1];
		flag = true;
	}
}

int main(){
	
	int n,m;
	cin>>n>>m;
	for(int i=0;i<n;i++){
		int a;
		cin>>a;
		push(a);
	}
	
	for(int i=0;i<m;i++){
		int a;
		cin>>a;
		flag = false;
		get_path(a);
		cout<<endl;
	}
	
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41700374/article/details/82019958