文章目录
二叉树的前中后三种遍历方式,各有各的特点。
我们可以通过前中后三种遍历方式中两种遍历方式的组合去还原一棵二叉树。
(无重复元素的树)
注意:
我们只能通过前中,中后两种方式去还原一棵确定的二叉树。
原因:
以前序遍历和中序遍历为例。
前序遍历是通过根左右的方式进行访问的,中序遍历时通过左根右的方式进行遍历的。
显而易见,前序遍历区分出根和左右节点,中序遍历区分出左节点和右节点,如此一来,根和左右节点的顺序一目了然,我们就可以通过代码来实现了。
前序遍历和中序遍历还原二叉树
思路(迭代)
- 构建根节点,创建一个索引指向中序遍历的首个元素,创建一个栈用来存储已经访问过的节点(此处的已经访问过指的是,将该节点的val进行赋值并存储,左右节点,并未完全构造),创建一个node指向栈顶元素。
- 对前序遍历的数组进行循环访问,当前栈顶元素 和中序遍历ord [index] (索引)值不相等时说明,当前的pre [i] 并不是这棵树的最左节点,于是创建新节点,为新节点赋值,并将新节点入栈,让栈顶元素的左子树指向这个新节点,并且更新栈顶元素。
- 当 栈顶元素的值 与 ord [index] 相等时说明,当前 前序遍历已经访问到了最左元素,于是循环将index后移并更新node然后进行弹栈操作,(注意 此处必须先更新栈顶元素node,然后进行弹栈操作,因为当跳出循环后,node的右孩子需要被指向。node的作用就是为了让节点和左右孩子进行相连)。
- 循环结束,返回根节点即可。
代码
class Solution {
public:
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
if (!preorder.size()) {
//条件判断
return nullptr;
}
TreeNode* root = new TreeNode(preorder[0]);//建立根节点
stack<TreeNode*> stk;