ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

树的基础知识2-二叉树的构造

2021-03-06 16:31:18  阅读:181  来源: 互联网

标签:pre index preorder int 构造 基础知识 二叉树 post inorder


1. 从中序和后序遍历构造二叉树

力扣106
题目描述:根据一棵树的中序遍历与后序遍历构造二叉树。你可以假设树中没有重复的元素。
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/
9 20
/
15 7

class Solution {
    int[] inorder;//因为在creatTree方法中用到这两个数组变量了
    int[] preorder;
    Map<Integer, Integer> map = new HashMap<>();

    //前序负责找根,中序负责分割
    public TreeNode creatTree(int ol, int or, int pl, int pr){
        if(ol > or) return null;
        TreeNode root = new TreeNode(preorder[pl]);
        int index = map.get(preorder[pl]);
        root.left = creatTree(ol, index - 1, pl + 1,pl + index - ol);
        root.right = creatTree(index +1, or, pl + index - ol +1 , pr);
        return root;
    }

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        this.inorder = inorder;
        this.preorder = preorder;
        //先把中序加入哈希表中
        for(int i = 0; i < inorder.length; i++){
            map.put(inorder[i], i);
        }
        return creatTree(0, inorder.length - 1, 0, preorder.length - 1);
    }
}
2. 从前序和中序构建二叉树

力扣105,剑指offer7
题目要求:输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/
9 20
/
15 7

class Solution {
    int[] inorder;//因为在creatTree方法中用到这两个数组变量了
    int[] preorder;
    Map<Integer, Integer> map = new HashMap<>();

    //前序负责找根,中序负责分割
    public TreeNode creatTree(int ol, int or, int pl, int pr){
        if(ol > or) return null;
        TreeNode root = new TreeNode(preorder[pl]);
        int index = map.get(preorder[pl]);
        root.left = creatTree(ol, index - 1, pl + 1,pl + index - ol);
        root.right = creatTree(index +1, or, pl + index - ol +1 , pr);
        return root;
    }

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        this.inorder = inorder;
        this.preorder = preorder;
        //先把中序加入哈希表中
        for(int i = 0; i < inorder.length; i++){
            map.put(inorder[i], i);
        }
        return creatTree(0, inorder.length - 1, 0, preorder.length - 1);
    }
}
3.由前序和后序构建二叉树

力扣889
题目描述:返回给定的前序和后序遍历匹配的任何二叉树。(不唯一,任何一个答案即可)
pre 和 post 遍历中的值是不同的正整数。
输入:pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1]
输出:[1,2,3,4,5,6,7]

class Solution {
    int[] pre;
    int[] post;
    Map<Integer, Integer> map = new HashMap<>();

    public TreeNode creatTree(int pre_l, int pre_r, int post_l, int post_r){
        if(pre_l > pre_r) return null;
        TreeNode root = new TreeNode(pre[pre_l]);
        if(pre_l == pre_r) return root;  //数组要提前结束,否则后面pre[pre_l+1]数组会溢出越界
        //前序根节点后面的这个节点一定是下一层的一个小根节点,用来分割后序(与其他两题的区别)
        int index = map.get(pre[pre_l + 1]);
        root.left = creatTree(pre_l + 1, pre_l+1+index-post_l, post_l, index);
        root.right = creatTree(pre_l+1+index-post_l+1, pre_r, index+1, post_r);
        return root;
    }

    public TreeNode constructFromPrePost(int[] pre, int[] post) {
        this.pre = pre;
        this.post = post;
        for(int i = 0;i < post.length; i++){
            map.put(post[i], i);
        }
        return creatTree(0, pre.length - 1, 0, post.length - 1);
    }
}

总结 1-3 : 三种遍历两两结合,重构二叉树的过程。
只有在前后序结合时,需要考虑一下数组越界的问题(代码中已指出)
其他两种只需要正常考虑边界范围即可

标签:pre,index,preorder,int,构造,基础知识,二叉树,post,inorder
来源: https://blog.csdn.net/Aylwang/article/details/114445133

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有