剑指 Offer 07. 重建二叉树
要求:输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
首先复习一下通过前中后序遍历一棵树:
- 前序:根节点-左子树-右子树;
- 中序:左子树-根节点-右子树;
- 后序:左子树-右子树-根节点;
通过图来解释:
- 前序遍历:ABDEGHCF
- 中序遍历:DBGEHACF
- 后序遍历:DGHEBFCA
bia上一张前辈博文里的图:
下面是前中后序遍历的代码:
void PreTravel(BinaryTree B)
{
if(B)
{
cout<<B->data<<" ";
PreTravel(B->left);
PreTravel(B->right);
}
}
void MidTravel(BinaryTree B)
{
if(B)
{
MidTravel(B->left);
cout<<B->data<<" ";
MidTravel(B->right);
}
}
void PostTravel(BinaryTree B)
{
if(B)
{
PostTravel(B->left);
PostTravel(B->right);
cout<<B->data<<" ";
}
}
言归正传,回归本题:
思路:
- 先在前序数组中找到根节点,也就是第一个,然后在中序数组中找到根节点的位置;
- 根据根节点的位置,将中序数组划分为左子树和右子树;
- 根据在中序数组中划分结果,将前序数组也进行划分;
- 递归处理左右子树;
这位前辈博客中的图示也可以帮助大家理解:
根据前序和中序遍历序列来构建二叉树
C++的代码如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
//1、判断前序和中序遍历的数组是否为空
if(preorder.empty() || inorder.empty())
{
return nullptr;
}
//2、先在前序数组中找到根节点,也就是第一个,然后在中序数组中找到根节点的位置;注意这里使用auto关键字;
TreeNode *root = new TreeNode(*preorder.begin());
auto rootPosition_In_inorder = find(inorder.begin(),inorder.end(),*preorder.begin());
//3、根据根节点的位置,将中序数组划分为左子树和右子树
vector<int> left_In_inorder(inorder.begin(),rootPosition_In_inorder);
vector<int> Right_In_inorder(rootPosition_In_inorder+1,inorder.end());
//4、根据在中序数组中划分结果,将前序数组也进行划分
int num = left_In_inorder.size();
vector<int> left_In_preorder(preorder.begin()+1,preorder.begin()+1+num);
vector<int> Right_In_preorder(preorder.begin()+1+num,preorder.end());
//5、递归处理左右子树
root->left = buildTree(left_In_preorder,left_In_inorder);
root->right = buildTree(Right_In_preorder,Right_In_inorder);
return root;
}
};
对于auto的说明:
- C语言中使用auto定义的变量可以不予初始化,但在C++中必须初始化;
- auto的对应类型不是使用new出来的变量,而是static变量,static变量是程序接收的时候才释放对象的,但它不需要手动释放。
- auto可以在声明变量的时候根据变量初始值的类型自动为此变量选择匹配的类型。
python代码:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
# 1.先判断是否满足条件
if not preorder or not inorder:
return None
# 2.在前序数组中找到根节点,注意要从int强制转换为TreeNode型
root = TreeNode(preorder[0])
# 3.在中序数组中找到根节点的位置
rootIndex = inorder.index(preorder[0])
# 4.将前序数组和中序数组根据左右子树分开,并进行递归处理
root.left = self.buildTree(preorder[1:1+rootIndex],inorder[:rootIndex])
root.right = self.buildTree(preorder[1+rootIndex:],inorder[rootIndex+1:])
# 5.返回
return root