ACM - C++ 由字符数组创建二叉树并作中序遍历

题目描述

由提供的字符数组创建二叉树，并基于此进行中序遍历。注意：下面将介绍两种情况，即一种是基于二叉树的层序遍历数组，一种是基于二叉树的前序遍历数组。
例：

其中，‘#’代表为空。
上述二叉树的层序遍历结果为：{1, 2, 4, #, 3, 5, #}；
前序遍历结果为：{1, 2, #, 3, 4, 5, #};
中序遍历结果为：{2, 3, 1, 5, 4}
需要注意的是，可以通过 前序 + 中序 与 中序 + 后序 实现对二叉树中元素位置正确的确定，单纯靠一种遍历结果进行确定，可能会出现不同的结果。

代码实现

#pragma once
#include<iostream>
#include<vector>
using namespace std;

typedef char TDataType;
//二叉树节点  二叉链
typedef struct BNode
{
	BNode *_left;
	BNode *_right;
	TDataType _data;
	BNode(char data) :_data(data), _left(nullptr), _right(nullptr){}
};

// 按照层序遍历构建二叉树
BNode* createTree(const vector<char> arr)
{
	vector<BNode*> vn(arr.size(), nullptr);
	BNode* root = nullptr;
	for (int i = 0; i < arr.size(); i++)
	{
		BNode* node = nullptr;
		if (arr[i] != '#')
			node = new BNode(arr[i]);
		vn[i] = node;
		if (i == 0)
			root = node;
	}
	for (int i = 0; i * 2 + 1 < vn.size(); i++)
	{
		if (vn[i])
			vn[i]->_left = vn[2 * i + 1];
		if (2 * i + 2 < vn.size())
			vn[i]->_right = vn[2 * i + 2];
	}
	return root;
}

// 按照前序遍历构建二叉树
BNode* createTree2(const vector<char> arr, int *idx)
{
	if ((*idx) + 1 > arr.size())
		return nullptr;
	if (arr[*idx] == '#')
	{
		(*idx)++;
		return nullptr;
	}
	BNode* root = new BNode(arr[*idx]);
	(*idx)++;
	root->_left = createTree2(arr, idx);
	root->_right = createTree2(arr, idx);
	return root;
}

void preOrder(BNode* root)
{
	if (root)
	{
		cout << root->_data << " ";
		preOrder(root->_left);
		preOrder(root->_right);
	}
}

void inOrder(BNode* root)
{
	if (root)
	{
		inOrder(root->_left);
		cout << root->_data << " ";
		inOrder(root->_right);
	}
}

int main()
{
	cout << "基于层序遍历：" << endl;
	vector<char> arr = { '1', '2', '4', '#', '3', '5', '#' };// 一个二叉树层序遍历的结果
	BNode* root = createTree(arr);
	inOrder(root); cout << endl;
	//preOrder(root); cout << endl;

	cout << endl << "基于前序遍历：（注意有问题）：" << endl;
	vector<char> arr2 = { '1', '2', '#', '3', '4', '5', '#' };//前序遍历的结果
	int idx = 0;
	BNode* root2 = createTree2(arr2, &idx);
	inOrder(root2); cout << endl << endl;
	
	system("pause");
	return 0;
}

结果分析

由图可知，基于层序遍历进行二叉树建立时可获得正确结果，但单纯用前序进行构建时，则会出错。上图中基于前序建立的二叉树结构如下，很明显与一开始给定的二叉树结构不同！