D. DS查找—二叉树平衡因子

目录

题目描述

思路分析

AC代码

---

题目描述

二叉树用数组存储，将二叉树的结点数据依次自上而下,自左至右存储到数组中，一般二叉树与完全二叉树对比，比完全二叉树缺少的结点在数组中用0来表示。

计算二叉树每个结点的平衡因子，并按后序遍历的顺序输出结点的平衡因子。

--程序要求--

若使用C++只能include一个头文件iostream；若使用C语言只能include一个头文件stdio

程序中若include多过一个头文件，不看代码，作0分处理

不允许使用第三方对象或函数实现本题的要求

输入

测试次数t

每组测试数据一行，数组元素个数n，后跟n个字符，二叉树的数组存储。

输出

对每组测试数据，按后序遍历的顺序输出树中结点的平衡因子（测试数据没有空树）

输入样例 

2
6 ABC00D
24 ABCD0EF0000H00000000000I

输出样例

 B 0
D 0
C 1
A -1
D 0
B 1
I 0
H 1
E 2
F 0
C 2
A -2

思路分析

这道题二叉树使用数组存储的，就是根结点放在数组下标为1的地方。第i个结点的左孩子是2*i,右孩子是2*i。创建三个数组。数组data[]放结点。数组depth[]存以该结点为根结点的树的深度。数组balance[]存该结点的平衡因子。

初始化depth为0.然后倒着看，从最后一个结点开始。每个结点的深度都等于max(左孩子深度，右孩子深度)+1。

然后平衡因子balance[]就可以正着算。balance[i]=depth[2*1]-depth[2*i+1];

中序遍历是一个递归。

AC代码

#include<iostream>
using namespace std;
int max(int a, int b) {
	if (a > b) return a;
	else return b;
}
void display(int i, char *c, int *b) {
	if (c[2 * i] != '0') {
		display(2 * i, c, b);
	}
	if (c[2 * i + 1] != '0') {
		display(2 * i + 1, c, b);
	}
	cout << c[i] << " " << b[i] << endl;
}
int main() {
	int t, n;
	cin >> t;
	while (t--) {
		cin >> n;
		char *data = new char[1000];
		for (int i = 0; i < 1000; i++) {
			data[i] = '0';
		}
		int *depth = new int[1000];
		for (int i = 0; i < 1000; i++) {
			depth[i] = 0;
		}
		int *balance = new int[1000];
		for (int i = 0; i < 1000; i++) {
			balance[i] = 0;
		}
		for (int i = 1; i <= n; i++) {
			cin >> data[i] ;
		}
		for (int i = n; i >= 1; i--) {
			if (data[i] == '0') depth[i] = 0;
			else  depth[i] = max(depth[2 * i], depth[2 * i + 1]) + 1;
		}
		for (int i = 1; i <= n; i++) {
			balance[i] = depth[2 * i] - depth[2 * i + 1];
		}
		display(1, data, balance);
	}
	return 0;
}

有错误欢迎指正！