二叉树--1-重建二叉树

题目

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

思路

用递归实现：

1）鲁棒性，判断是输入pre与vin是否合法

2）存pre的第一个元素为根节点；

3）判断根节点在vin中的位置；

4）获得left_pre    left_vin   right_pre   right_vin

5）return head 递归2-4步骤

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

struct TreeNode
{
	int value;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : value(x), left(nullptr), right(nullptr) {}
};

//preorder  infix order  postorder
class Solution
{
public:
	TreeNode * restuctBinaryTree(vector<int> pre, vector<int> vin)
	{
		if (pre.size() == 0 || vin.size() == 0 || pre.size() != vin.size())
			return nullptr;
		TreeNode * head = new TreeNode(pre[0]);
		int root = 0, i;
		for (i = 0; i < vin.size(); ++i)			//在中序中找根节点
		{
			if (vin[i] == pre[0])
			{
				root = i;
				break;
			}
		}
		vector<int> left_pre, right_pre, left_vin, right_vin;
		for (i = 0; i < root; ++i)
		{
			left_pre.push_back(pre[i + 1]);
			left_vin.push_back(vin[i]);
		}
		for (int i = root + 1; i < vin.size(); ++i)
		{
			right_pre.push_back(pre[i]);
			right_vin.push_back(vin[i]);
		}
		head->left = restuctBinaryTree(left_pre, left_vin);
		head->right = restuctBinaryTree(right_pre, right_vin);
		return head;
	}
};

int main()
{
	Solution s;
	vector<int> pre{ 1, 2, 4, 7, 3, 5, 6, 8 };
	vector<int> vin{ 4, 7, 2, 1, 5, 3, 8, 6 };
	TreeNode *node = s.restuctBinaryTree(pre, vin);
	cout << node->value << node->left->value << node->left->left->value <<endl;
	system("pause");
	return 0;
}