天天看點

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

###思路:

本題以數組存儲最小堆。依次插入,核心代碼在 resort 函數。每次插入一個元素,放在數組最後,即樹層次周遊的最後一個位置,然後和他的父親節點做比較,如果它大于父親節點的值,則滿足最小堆的定義,否則将它和父親節點做交換,循環直至到根節點或者小于父親節點值。

最小堆:根節點的值比左右節點的值都小的二叉樹,也叫小頂堆/小根堆。

最大堆:根節點的值比左右節點的值都大的二叉樹,也叫大頂堆/大根堆。

#include<stdio.h>
#define N 1002
int h[N]={0};
void resort(int i);
void way2root(int i);
int main(){
	int n, m;
	scanf("%d%d",&n, &m);
	for(int i = 1; i <= n; i++){//二叉樹從1開始編碼 
		scanf("%d",h+i);
		resort(i);
	} 
	while(m--){
		int index;
		scanf("%d",&index);
		way2root(index);
	}
	return 0;
}

void resort(int i){
	//把新插入的元素,放入到數組最後一個元素
	//然後重新組織成最小堆
	while(i/2&& h[i/2] > h[i]){
		int tmp = h[i];
		h[i] = h[i/2];
		h[i/2] = tmp;
		i /= 2;
	} 
}
void way2root(int i){
	while(i != 1){
	    printf("%d ",h[i]);
		i /= 2;	
	}
	printf("%d\n",h[1]);
}