C++实现树 - 04 二叉树的构建（数组）

数据结构与算法专栏 —— C++实现

写在前面：
通过前面两讲的学习，大家可能对二叉树有了比较深的感悟，但可能会发现一个小问题，我们在构建二叉树的时候都是一个个插入的，非常的不方便。那么这节课我们就来看看，如何通过输入一个数组来快速构建起一个二叉树。这里会介绍通过顺序数组、前序数组和后序数组如何构建二叉树。

完全二叉树的构建

在前面的学习中，我们知道了完全二叉树有一个非常 nice 的性质，它每个结点的左孩子在数组中的位置等于 2x ，右孩子等于 2x+1 。通过这个性质，我们可以快速的构建起一个二叉树。

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

struct Node {
	int data;
	Node *left_node;
	Node *right_node;
};

int n;	//表示数组长度

//递归法构建二叉树
Node *creatTree(vector<int> nums, int index) {
	//判断是否为空
	if (nums[index] == '#')
		return NULL;

	//创建新结点
	Node *root = new Node;
	root->data = nums[index];

	//设置左右指针
	if (index * 2 <= n)
		root->left_node = creatTree(nums, index * 2);	//找到左孩子
	else
		root->left_node = NULL;
	if (index * 2 + 1 <= n)
		root->right_node = creatTree(nums, index * 2 + 1);	//找到右孩子
	else
		root->right_node = NULL;

	return root;
}

//非递归构建二叉树（利用栈）—— 前序遍历
void stackCreatTree(vector<int> nums) {
	int n = nums.size();
	stack<int> s;
	int index;	//记录遍历到第几个元素
	s.push(1);	//推入根结点，开始遍历
	while (!s.empty()) {
		index = s.top();
		s.pop();
		cout << nums[index] << " ";
		//先推入右孩子
		if (index * 2 + 1 < n && nums[index * 2 + 1] != '#')
			s.push(index * 2 + 1);
		//再推入左孩子，因为是栈结构，先进后出，而且前序遍历需要优先访问所有左子树，故只有一直放在栈顶才能先访问
		if (index * 2 < n && nums[index * 2] != '#')
			s.push(index * 2);
	}
	cout << endl;
}

//前序遍历
void PreOrderTraverse(Node *root) {
	if (root) {
		cout << root->data << " ";
		PreOrderTraverse(root->left_node);
		PreOrderTraverse(root->right_node);
	}
}

int main() {
	vector<int> nums = { '#', 1, 2, 3, 4, 5, '#', 6, '#', '#', 7, 8 };	//'#'指向空，并且从下标1开始存数据
	n = nums.size();
	Node *root = creatTree(nums, 1);
	PreOrderTraverse(root);
	cout << endl;
	stackCreatTree(nums);
	return 0;
}

根据前序和后序数组构建二叉树

除了给定一个顺序存储的数组外，还可能会给定前序遍历的结果，让你构建出二叉树。这里的操作和上面比较类似，但区别是我们获取结点的左右孩子的方式变了，上面我们是根据完全二叉树的性质。
（1）这里我们可以根据前序遍历的性质，也就是遍历结果遵循 根结点 | 左子树 | 右子树 的结果。所以，我们直接可以用一个全局变量下标来记录我们遍历的位置，从头往后遍历，每次都加 1 ，这样就可以连起来了。
（2）同样，后序遍历结果是遵循 左子树 | 右子树 | 根结点 的结果。所以，我们可以从最后一个元素开始往前遍历，每次都减 1 ，同样可以构建出来。

这里我们不用去判断下标是否超过数组长度，因为给定的数组是一定满足要求的即左子树右子树是否为空直接看’#'就可以。

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

struct Node {
	int data;
	Node *left_node;
	Node *right_node;
};

int n;	//表示数组长度
int index;	//表示遍历到的下标

//前序数组创建二叉树

Node *prevCreatTree(vector<int> nums) {
	//判断是否为空
	if (nums[index] == '#') {
		index++;
		return NULL;
	}

	//创建结点
	Node *root = new Node;
	root->data = nums[index++];
	root->left_node = prevCreatTree(nums);
	root->right_node = prevCreatTree(nums);

	return root;
}

//后序数组创建二叉树
int cnt;
Node *backCreatTree(vector<int> nums) {
	if (nums[cnt] == '#') {
		cnt--;
		return NULL;
	}
	Node *root = new Node;
	root->data = nums[cnt--];
	root->right_node = backCreatTree(nums);
	root->left_node = backCreatTree(nums);
	return root;
}

//中序遍历
void InOrderTraverse(Node *root) {
	if (root) {
		InOrderTraverse(root->left_node);
		cout << root->data << " ";
		InOrderTraverse(root->right_node);
	}
}


int main() {
	//'#'指向空
	vector<int> nums1 = { 1, 2, 3, '#', '#', 4, '#', '#', 5, '#', '#' };
	vector<int> nums2 = {'#', '#', 3, '#', '#', 4, 2, '#', '#', 5, 1};
	index = 0;	//初始化下标
	Node *prev_root = prevCreatTree(nums1);
	InOrderTraverse(prev_root);
	cout << endl;
	cnt = nums2.size() - 1;
	Node *back_root = backCreatTree(nums2);
	InOrderTraverse(back_root);
	return 0;
}

如果大家有什么问题的话，欢迎在下方评论区进行讨论哦~

【上一讲】树 - 03 二叉排序树
【下一讲】树 - 05 线索二叉树