码迷,mamicode.com
首页 > 其他好文 > 详细

从前序与中序遍历序列构造二叉树

时间:2020-11-13 12:22:54      阅读:7      评论:0      收藏:0      [点我收藏+]

标签:from   ==   遍历   根据   vat   rom   tree   解题思路   problems   

从前序与中序遍历序列构造二叉树

题目:
根据一棵树的前序遍历与中序遍历构造二叉树。

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

例如,给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
技术图片

解题思路:前序数组的第一个元素就是根节点,在中序数组中找到这一个元素,然后该元素左边就是左子树,右边是右子树,再分别对这两个部分进行构造

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return build(preorder, inorder, 0, preorder.length - 1, 0, preorder.length - 1);
    }
    
    private TreeNode build(int[] preorder, int[] inorder, int preL, int preR, int inL, int inR) {
        if(preL > preR || inL > inR) return null;
        
        int target = preorder[preL];
        int cur = inL;
        for(int i = cur; i <= preR; i++) {
            if(inorder[i] == target) {
                cur = i;
                break;
            }
        }
        
        TreeNode node = new TreeNode(target);
        //为什么是preL + cur - inL
        //cur - inL就是中序数组的左子树部分节点数
        node.left = build(preorder, inorder, preL + 1, preL + cur - inL, inL, cur - 1);
        node.right = build(preorder, inorder, preL + cur - inL + 1, preR, cur + 1, inR);
        
        return node;
    }
}

从前序与中序遍历序列构造二叉树

标签:from   ==   遍历   根据   vat   rom   tree   解题思路   problems   

原文地址:https://www.cnblogs.com/katoMegumi/p/13914599.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!