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;
}