05-樹7 堆中的路徑(c/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

思路

首先根據所給的元素構建出小頂堆，堆的構建是層序依次放置，可以選擇用數組來保存數據，比較葉子結點與根節點的大小，若小於根則交換。

根節點序號從1開始，設置H[0]=-10001作爲邊界值

代碼

#include<bits/stdc++.h> 
using namespace std;

int main(){
	int n,m;//插入小頂堆的整數的個數，m個要打印的路徑 
	cin>>n;
	cin>>m;
	int H[n+1],sub[m];
	H[0]=-10001;
	int temp,a;
	for(int i=1;i<=n;i++){
		cin>>H[i];
		a=i;
		do{
			if(H[a]<H[a/2]){
				temp=H[a];
				H[a]=H[a/2];
				H[a/2]=temp;  
			}
			else{
				break;
			}
			a=a/2;
		}while(a/2);//將插入的值與根節點比較，若小於根則交換 
	} 
	for(int i=0;i<m;i++){
		cin>>sub[i];
	}
	for(int i=0;i<m;i++){
		for(int j=sub[i];j>0;j=j/2){
			cout<<H[j];
			if(j!=1){
				cout<<" ";
			}
		}
		if(i<m-1){
			cout<<endl;
		}
	}
	return 0;
}