1. 程式人生 > 實用技巧 >DS二叉樹——二叉樹之陣列儲存

DS二叉樹——二叉樹之陣列儲存

題目描述

二叉樹可以採用陣列的方法進行儲存,把陣列中的資料依次自上而下,自左至右儲存到二叉樹結點中,一般二叉樹與完全二叉樹對比,比完全二叉樹缺少的結點就在陣列中用0來表示。,如下圖所示

從上圖可以看出,右邊的是一顆普通的二叉樹,當它與左邊的完全二叉樹對比,發現它比完全二叉樹少了第5號結點,所以在陣列中用0表示,同樣它還少了完全二叉樹中的第10、11號結點,所以在陣列中也用0表示。

結點儲存的資料均為非負整數

輸入

第一行輸入一個整數t,表示有t個二叉樹

第二行起,每行輸入一個數組,先輸入陣列長度,再輸入陣列內資料,每個資料之間用空格隔開,輸入的資料都是非負整數

連續輸入t行

輸出

每行輸出一個示例的先序遍歷結果,每個結點之間用空格隔開

樣例輸入

3
3 1 2 3
5 1 2 3 0 4
13 1 2 3 4 0 5 6 7 8 0 0 9 10

樣例輸出

1 2 3
1 2 4 3
1 2 4 7 8 3 5 9 10 6

提示

注意從陣列位置和二叉樹深度、結點位置進行關聯,或者父子結點在陣列中的位置存在某種管理,例如i, i+1, i/2, i+1/2........或者2i, 2i+1.......仔細觀察哦

思路

這道題我做了很久,用了笨辦法,就是先輸出左邊再輸出右邊,單純記錄一下程式碼。

#include<iostream>
#include<math.h>
using namespace std;

int getLayer(int n) {
	int j = 0;
	while (n / 2 != 0) {
		j++;
		n = n / 2;
	}
	return j;
}

int main() {
	int t;
	cin >> t;
	while (t--) {
		int len;
		cin >> len;
		int* num = new int[len];
		for (int i = 0; i < len; i++) {
			cin >> num[i];
		}
		int layer = getLayer(len);
		int x, n, j = 0;
		cout << num[0] << ' ';
		for (int i = 1; i <= layer; i++) {
			n = pow(2, i);
			if (num[n - 1]) {
				cout << num[n - 1] << ' ';
				num[n - 1] = 0;
			}
		}
		if (num[n] && n != 2)
			cout << num[n] << ' ';
		if (layer > 2) {
			for (int i = 2; i <= layer; i++) {
				n = pow(2, i);
				x = pow(2, i - 1);
				int j = n + pow(2, i - 2);
				while (j <= n + x - 1) {
					if (num[j - 1]) {
						cout << num[j - 1] << ' ';
						num[j - 1] = 0;
					}
					j++;
				}
			}
		}
		for (int i = 1; i <= layer; i++) {
			n = pow(2, i);
			x = pow(2, i - 1);
			if (num[n + x - 1] > 0 && (n + x - 1) < len) {
				cout << num[n + x - 1] << ' ';
				num[n + x - 1] = 0;
			}
		}
		if (num[n + x] > 0 && (n + x) < len)
			cout << num[n + x] << ' ';
		if (layer > 2) {
			for (int i = 2; i <= layer; i++) {
				n = pow(2, i);
				int j = 2 * n - pow(2, i - 2);
				while (j <= 2 * n - 1 && (2 * n - 1) < len) {
					if (num[j - 1] > 0) {
						cout << num[j - 1] << ' ';
						num[j - 1] = 0;
					}
					j++;
				}
			}
		}
		cout << endl;
	}
	return 0;
}