《剑指offer》—4、重建二叉树

 题目描述

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

解题思路：

1、有关树或图的算法题，看到很容易把人吓住，其实本质上不难，但相关知识点比较容易遗忘，回顾一下。

前序遍历：根左右

中序遍历：左根右
后序遍历：左右根

个人采用简单的记忆是根节点位置在哪就是哪种遍历，比如，前序遍历，根节点最前面；中序遍历，根节点中间；后序遍历，根节点最后。先左后右的相对位置，这样就容易记住了。

2、记住树一般用递归方式解题的偏多，知道前序遍历和中序遍历后，如何还原二叉树，具体针对上述题目来分析，

1）、前序遍历的第一个节点为根节点，由题中前序遍历可知，1为根节点；

2）、中序遍历的根节点前面的节点均为左子树的节点，所以左子树上的节点为{4,7,2}；

3）、由中序遍历序列可知，右子树节点为{5,3,8,6}；

一般你知道这些信息就足够了，后面肯定也是按照上述规律进行递归的。

已知题目给出的函数原型如下

struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in)

根据递归，简单描述：

1）、确定根，确定左子树，确定右子树。根节点node为前序遍历的第一个数；即

TreeNode *node = new TreeNode(1);

2 )、左子树中递归。即

node->left = reConstructBinaryTree({ 2,4,7 }, { 4,7,2 });

3 )、右子树中递归。即

node->right = reConstructBinaryTree({ 3,5,6,8 }, { 5,3,8,6 });

递归建树，直至为空。

是不是觉得也不难啊，再看下面的参考代码就立马通俗易懂了

参考代码：

/*************************************************
Copyright:牛客网在线编程《剑指offer》
Author:zhouyuan
Date:2017-02-24
Description:重建二叉树
**************************************************/

#include <vector>
using namespace std;

//Definition for binary tree
struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
	TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> in) {
		if (pre.empty() || in.empty()) {
			return NULL;
		}

		int val = pre.at(0);
		TreeNode *node = new TreeNode(val);
		int size = in.size();
		int index = 0;

		for (; index < size; index++) {
			if (in.at(index) == val) {
				break;
			}
		}

		vector<int> pre1, in1, pre2, in2;
		for (int i = 1; i <= index; i++) {
			pre1.push_back(pre.at(i));
		}
		for (int i = 0; i < index; i++) {
			in1.push_back(in.at(i));
		}
		for (int i = index + 1; i < size; i++) {
			pre2.push_back(pre.at(i));
			in2.push_back(in.at(i));
		}

		node->left = reConstructBinaryTree(pre1, in1);
		node->right = reConstructBinaryTree(pre2, in2);

		return node;
	}
};