美文网首页
106. 从中序与后序遍历序列构造二叉树

106. 从中序与后序遍历序列构造二叉树

作者: hqwer | 来源:发表于2019-12-23 19:51 被阅读0次
    题目描述:

    根据一棵树的中序遍历与后序遍历构造二叉树。

    注意:
    你可以假设树中没有重复的元素。

    例如,给出

    中序遍历 inorder = [9,3,15,20,7]
    后序遍历 postorder = [9,15,7,20,3]
    返回二叉树
    链接地址:https://leetcode-cn.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal/

    难度:中等
    解题思路:

    先复习了一下概念
    前序遍历:中左右
    中序遍历:左中右
    后序遍历:左右中
    后序遍历从后往前依次为中,在中序遍历中找到对应的位置,该位置的左边构成了左子树,右边构成了右子树,递归调用,直到中序遍历上的指针相遇。

    代码:
    class Solution {
        int index;
    
        public TreeNode buildTree(int[] inorder, int[] postorder) {
            index = postorder.length - 1;
            return fun(inorder, postorder, 0, inorder.length - 1);
        }
    
        private TreeNode fun(int[] inorder, int[] postorder, int start, int end) {
            if (start > end) return null;
    
            int val = postorder[index--];
            for (int i = start; i <= end; i++) {
                if (inorder[i] == val) {
                    TreeNode root = new TreeNode(val);
                    root.right = fun(inorder, postorder, i + 1, end);
                    root.left = fun(inorder, postorder, start, i - 1);
                    return root;
                }
            }
            return null;
        }
    }
    

    注意这里的root.right和root.left递归调用顺序不能乱。因为后序遍历从右往左依次存放的是中,右中,右右中等等,从右往左依次拿出了右子树的中数。

    分析:

    时间复杂度O(NlogN),空间复杂度O(N),耗时34 ms。时间又有点长,排名百分之七十。
    学习用时最短代码:

    public TreeNode buildTree(int[] inorder, int[] postorder) {
        return build(inorder, postorder, 0, postorder.length - 1, postorder.length);
    }
    
    public TreeNode build(int[] inorder, int[] postorder, int inStart, int postEnd, int length) {
        if (length == 0) return null;
        int root = postorder[postEnd];
        TreeNode treeNode = new TreeNode(root);
        if (length == 1) return treeNode;
        for (int i = length - 1; i >= 0; i--) {
            if (root == inorder[inStart + i]) {
                //注意这里的postEnd-length+i不等于i,数组长度变为子数组长度,i的下标已经改变,
                //i为左子树长度,在postorder中为左子树于右子树的分界点
                treeNode.left = build(inorder, postorder, inStart, postEnd - length + i, i);
                treeNode.right = build(inorder, postorder, inStart + i + 1, postEnd - 1,  length - 1 - i);
                return treeNode;
            }
        }
        return null;
    }
    

    只需要2 ms.....他的思路是在后序遍历中找出中值,然后先找左值,最后再找右值。这样做递归调用里面的for循环次数少了很多。所以用时很短。

    总结:

    二叉树的算法题还是比较薄弱,这思路也是想了很长时间才想到T__T。继续努力!!!

    相关文章

      网友评论

          本文标题:106. 从中序与后序遍历序列构造二叉树

          本文链接:https://www.haomeiwen.com/subject/nuygectx.html