给定一棵树的前序遍历 preorder
与中序遍历 inorder
。请构造二叉树并返回其根节点
法一:递归分治
对于任意一颗树而言,前序遍历的形式总是
[ 根节点, [左子树的前序遍历结果], [右子树的前序遍历结果] ]
即根节点总是前序遍历中的第一个节点。而中序遍历的形式总是
[ [左子树的中序遍历结果], 根节点, [右子树的中序遍历结果] ]
只要我们在中序遍历中定位到根节点,那么我们就可以分别知道左子树和右子树中的节点数目。由于同一颗子树的前序遍历和中序遍历的长度显然是相同的,因此我们就可以对应到前序遍历的结果中,对上述形式中的所有左右括号进行定位。
而中序序列的左右括号很容易定位,根据上一个左右括号和本次的节点位置就可以。
这样以来,我们就知道了左子树的前序遍历和中序遍历结果,以及右子树的前序遍历和中序遍历结果,我们就可以递归地对构造出左子树和右子树,再将这两颗子树接到根节点的左右位置。
struct TreeNode{
int val;
TreeNode* left;
TreeNode* right;
TreeNode (a):left=nuullptr,right=nuullptr{
val=a;
}
}
class Solution {
private: unordered_map<int,int> inorder_map;
public:
TreeNode* building(vector<int>& preorder,vector<int>& inorder,int preorder_left,int preorder_right,int inorder_left,int inorder_right){
//以前序遍历为基准,右超过左代表子树不存在了
if(preorder_left>preorder_right)return nullptr;
//记录本次先序遍历元素位置
int preorder_root_index= preorder_left;
//记录本次中序遍历元素位置
int inorder_root_index=inorder_map[preorder[preorder_root_index]];
//构造节点
TreeNode* node=new TreeNode(preorder[preorder_root_index]);
//求出左子树的大小,知道等会如何安排左子树先序序列的左右界 和右子树先序序列的左右界。
int left_subtree_size=inorder_root_index-inorder_left;
//根据基本原理,定位先序序列中,左子树左右中括号,以及中序序列中左子树左右中括号
node->left=building(preorder,inorder,preorder_left+1,preorder_left+left_subtree_size,inorder_left,inorder_root_index-1);
//根据基本原理,定位先序序列中,右子树左右中括号,以及中序序列中右子树左右中括号
node->right=building(preorder,inorder,preorder_left+left_subtree_size+1,preorder_right,inorder_root_index+1,inorder_right);
return node;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int n=inorder.size();
//构造哈希映射,有利于在常数时间内,定位中序遍历序列中元素
for(int i=0;i<n;i++){
inorder_map[inorder[i]]=i;
}
//building函数递归的构造节点
return building(preorder,inorder,0,n-1,0,n-1);
}
};