剑指Offer(四)-- 非递归解法重建二叉树

2022-02-15 14:11:23 浏览数 (1)

Part1前言

剑指Offer & LeetCode刷题仓库https://github.com/Damaer/CodeSolution 文档地址https://damaer.github.io/CodeSolution/ 刷题仓库介绍刷题仓库:CodeSolution 编程知识库https://github.com/Damaer/Coding 文档地址https://damaer.github.io/Coding/#/

Part2重建二叉树

1题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

2思路 & 解答

相信大家对使用递归解答这个问题,已经有所了解,这里不再赘述,所有的递归理论上都可以用栈模拟,那么我们如何用栈解答呢?

我们可以一开始创建一个栈,分别用两个指针执行前序遍历和中序遍历的第一个元素,先将前序遍历的第一个元素压入栈中,因为前序遍历的特性,第一个元素肯定是根节点。

  • 开始循环,对比栈顶的元素和中序遍历数组的元素
    • 2.1 把栈顶元素和中序遍历的元素对比,相等则弹出之后,继续对比下一个元素与当前的栈顶元素,直到不相等为止。(相等说明没有右节点,弹出以为着退出上一层)
    • 2.2 不相等的时候,需要把当前的元素作为右叶子节点,压入栈中。
    • 1 如果不相等,说明当前栈顶元素还有左子树,因为如果没有左子树的话,前序的第一个元素和中序的第一个元素应该相等。既然有左子树,那么前序遍历指针指向的元素就是它的左子树根节点,建立关系,压栈。
    • 2 如果相等,那么说明当前的栈顶元素已经没有左子树了。

上面的文字可能比较难懂,但是不紧要,下面图文说明:

首先我们有前序和中序遍历的数组,原树结构大致了解一下:

把当前的前序遍历的元素 1 先放到栈里面,这个肯定是根节点:

对比中序遍历第一个元素 4,和栈顶元素 1,不相等,那么说明有左子树,前序遍历的第 2 个元素 2 就是左子树节点,关联成左子树,压栈:

同样的不相等,会持续压栈:

直到,中序遍历的第一个元素 4,已经等于栈顶元素 4,说明4 没有左子树了,因为 4 是在中序遍历里面,中序遍历完根节点,剩下的部分只能是右子树。

那么把 4 弹出去,中序遍历指针移动到下一个位置:

这个时候,7 肯定是之前节点 4 的右子树节点,那么关联关系之后,压入栈里面:

此时,结束了一次循环,注意前序遍历的指针会往后移动一位。

再次循环的时候,依然判断中序遍历中的数值是否等于栈顶元素,发现都是7,相等。弹出,移动到下一个位置,相当于退出了上一层:

依旧 2==2 相等,再次弹出:

同样中序遍历的 1 还是等于栈顶的 1,弹出,移动到下一位:

这个时候,栈顶元素的 1 已经被取出来了,说明左子树全部遍历完成了,剩下的部分是它的右子树内容了,那么前序遍历中,3就必定是根节点1的右子树的根,压入栈中,前序遍历索引指向下一个元素:

到这里其实是结束了第二轮的循环。

再次循环,判断中序遍历的数值和栈顶元素不相等,那么说明是左子树,前序遍历中的 5 压入栈内,索引后移:

中序遍历的数值和栈顶元素一对比,发现相等,说明5没有左子树了,弹出,索引后移:

依然 两个都是 3(说明 3 的左子树被遍历完成了,剩下的是 3 的右子树了),继续弹出,后移

此时,3 是刚刚弹出的元素,剩下的元素都是它的右子树,那么前序遍历中指向的数组6就是3的右子树,6 压入栈中:

对比栈顶元素6 和中序遍历中的8 发现不相等,那么把前序遍历中的 8 压栈,成为左子树:

对比栈顶元素 8 和中序遍历的 8 ,相等则弹出:

还是相等,继续弹出:

栈里面没有元素,并且数组都遍历结束,整个过程结束。

Java 代码实现如下:

代码语言:javascript复制
import java.util.Stack;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}
public class Solution7 {
    public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
        // 判空
        if (pre == null || pre.length == 0 || in == null || in.length == 0) {
            return null;
        }
        Stack<TreeNode> stack = new Stack<>();
        int preIndex = 0;
        int inIndex = 0;

        TreeNode root = new TreeNode(pre[preIndex]);
        stack.push(root);

        for (preIndex = 1; preIndex < pre.length; preIndex  ) {
            TreeNode node = stack.peek();
            // 不相等说明还有左子树
            if (node.val != in[inIndex]) {
                // 关联成为左子树,压栈
                node.left = new TreeNode(pre[preIndex]);
                stack.push(node.left);
            } else {
                // 相等说明,当前节点没有左子树
                while (!stack.isEmpty() && stack.peek().val == in[inIndex]) {
                    // 只要两者相等,说明没有右子树,弹出节点,退到上一层
                    node = stack.pop();
                    inIndex  ;
                }
                // 有右子树,关联
                node.right = new TreeNode(pre[preIndex]);
                stack.push(node.right);
            }
        }
        return root;
    }
}

C 代码如下:

代码语言:javascript复制
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) {
        // 判空
        if (pre.size() == 0 || vin.size() == 0) {
            return NULL;
        }
        stack<TreeNode *> stack;
        int preIndex = 0;
        int inIndex = 0;

        TreeNode *root = new TreeNode(pre[preIndex]);
        stack.push(root);

        for (preIndex = 1; preIndex < pre.size(); preIndex  ) {
            TreeNode *node = stack.top();
            // 不相等说明还有左子树
            if (node->val != vin[inIndex]) {
                // 关联成为左子树,压栈
                node->left = new TreeNode(pre[preIndex]);
                stack.push(node->left);
            } else {
                // 相等说明,当前节点没有左子树
                while (stack.size() != 0 && stack.top()->val == vin[inIndex]) {
                    // 只要两者相等,说明没有右子树,弹出节点,退到上一层
                    node = stack.top();
                    stack.pop();
                    inIndex  ;
                }
                // 有右子树,关联
                node->right = new TreeNode(pre[preIndex]);
                stack.push(node->right);
            }
        }
        return root;
    }
};

【作者简介】

秦怀,公众号【秦怀杂货店】作者,技术之路不在一时,山高水长,纵使缓慢,驰而不息。个人写作方向:Java源码解析,JDBC,Mybatis,Spring,Redis,分布式,剑指Offer,LeetCode等,认真写好每一篇文章,不喜欢标题党,不喜欢花里胡哨,大多写系列文章,不能保证我写的都完全正确,但是我保证所写的均经过实践或者查找资料。遗漏或者错误之处,还望指正。

平日时间宝贵,只能使用晚上以及周末时间学习写作

0 人点赞