题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路
1、首先需要理解前序中序后序的规则,前序是先根后左右,中序是先左再根再右,后序是先左右后跟。记忆的方法就是跟是在前还是在中还是在后。
2、又到了比较头疼的递归环节,我们只需要着眼于当前层。中需要把给的树,分成左右子树,然后把最后连接好的左右子树的根返回回去。
3、那么如何得到左右子树呢?这里有一点点小技巧,前序遍历的第一个节点,就是根节点,前序的左子树是从1到n的位置,n+1取不到,中序遍历根节点的位置左边就是所要构建的左子树,n的位置为根节点,n+1到最后为右子树。
做题可能出现的问题
python可能出现的问题
1、因为是树的问题,肯定是需要建立TreeNode的,
2、给的pre和tin都是数组,可以利用index找到根节点的位置,前提是里面不含有重复的数字。
3、递归的时候,要记得使用self因为这是在类里面。
C++可能出现的问题
1、TreeNode* node = new TreeNode(value);
2、int p需要写在外面面,而不能写在for的小括号里面,那样就变成局部变量了
3、c++没有python那么多语法糖,不能用slice的方式来,那么就用嵌套if语句来实现
python
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回构造的TreeNode根节点
def reConstructBinaryTree(self, pre, tin):
# write code here
if not pre:
return None
if not tin:
return None
root = TreeNode(pre[0])
n = tin.index(root.val)
root.left = self.reConstructBinaryTree(pre[1:n+1], tin[:n])
root.right = self.reConstructBinaryTree(pre[n+1:], tin[n+1:])
return root
C++
/**
* 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> vin) {
int in_size = vin.size();
if (in_size == 0)
{
return NULL;
}
vector<int> pre_left, pre_right, vin_left, vin_right;
int value = pre[0];
TreeNode* node = new TreeNode(value);
int p;
for(p=0; p<in_size;p++)
{
if (vin[p]==value)
break;
}
for (int i = 0; i<in_size; i++)
{
if(i<p)
{
vin_left.push_back(vin[i]);
pre_left.push_back(pre[i+1]);
}
else if(i>p)
{
vin_right.push_back(vin[i]);
pre_right.push_back(pre[i]);
}
}
node->left = reConstructBinaryTree(pre_left, vin_left);
node->right = reConstructBinaryTree(pre_right, vin_right);
return node;
}
};