class Solution {
public:
TreeNode* __buildTree(vector<int>& preorder,int pl,int pr, vector<int>& inorder,int il,int ir)
{
if (pl == pr) return nullptr;
TreeNode* root = new TreeNode(preorder[pl]);
int lsize = 0, rsize = 0;
for (int i = il; i < ir; i++) {
if (inorder[i] == preorder[pl]) {
lsize = i - il;
rsize = ir - i - 1;
if (lsize) root->left = __buildTree(preorder, pl + 1, pl + 1 + lsize, inorder, il, i);
if (rsize) root->right = __buildTree(preorder, pl + lsize + 1, pr, inorder, i + 1, ir);
return root;
}
}
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
return __buildTree(preorder,0,preorder.size(),inorder,0,inorder.size());
}
};
-
buildTree 方法:这是公开的接口,外部调用者会使用这个方法来开始重建二叉树的过程。它接受两个
vector<int>
类型的参数,分别代表前序遍历数组和中序遍历数组。然后,它调用__buildTree
方法,并传递必要的参数来开始重建过程。 -
__buildTree 方法:这是一个私有辅助方法,用于递归地重建二叉树。它接受四个参数:
preorder
:前序遍历数组。pl
和pr
:前序遍历数组的左右边界索引,用于定义当前子树的前序遍历范围。inorder
:中序遍历数组。il
和ir
:中序遍历数组的左右边界索引,用于定义当前子树的中序遍历范围。
这个方法首先检查前序遍历的左边界pl
是否等于右边界pr
。如果是,那么说明当前子树为空,因此返回nullptr
。
接下来,它创建一个新的树节点,其值为前序遍历数组中的当前元素(preorder[pl]
)。
然后,它遍历中序遍历数组,以找到与前序遍历数组中当前元素相同的元素。这个元素的索引在中序遍历数组中的位置将当前子树分为左子树和右子树。
接下来,它计算左子树和右子树的大小(lsize
和rsize
),然后递归地调用__buildTree
方法来重建左子树和右子树。这些子树的前序和中序遍历范围基于当前元素的索引进行调整。
最后,它将左子树和右子树分别连接到当前创建的根节点上,并返回根节点。
这个算法利用了前序遍历和中序遍历的特性来重建二叉树。前序遍历的第一个元素总是根节点,而在中序遍历中,根节点将数组分为左子树和右子树。通过找到根节点在中序遍历中的位置,我们可以确定左子树和右子树的范围,并递归地重建它们。