Construct Binary Tree from Preorder and Inorder Traversal

https://oj.leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/


Given preorder and inorder traversal of a tree, construct the binary tree.

Note:
You may assume that duplicates do not exist in the tree.

public TreeNode buildTree(int[] preorder, int[] inorder)


这一题利用到了一个特性是:一棵树的左子树的节点是不变的.....嗯,我在说什么呢?

其实是这样的:虽然preoreder和inorder的顺序肯定是不一样的。但是根节点左子树的范围和右边的范围是一样的。毕竟不管是in order还是pre order,都是先跑完左子树 + 根节点 再跑右子树的。所以如果说如果左子树的节点数目加上根节点的数目是i,那么不管是preorder还是inorder数组,前面i个节点都表示了左子树+根节点。

然后我们基于这个特性,再找到preorder和inorder的不同点。 preorder的第一个节点就是根节点,inorder要跑完所有左子树的节点才跑根节点。根据以上两个特质,做法如下:

1. 选取当前preorder的第一个元素(或者子集里的第一个元素)。

2.找到这个元素在inorder里对应的位置。譬如说这个位置是 i, 那么前面i - 1个位置就是这棵树的左子树了。同样的,preorder 在第一个元素的后面的i - 1个位置也都是左子树的内容。

3. 然后在inorder里以这个i为界,作为当前的根节点,然后分成左右两个部分,分别就是左子树和右子树的内容。同样,我们也可以根据preorder的特性找到左子树和右子树在数组里的范围。

4. 以上面3步找到的左子树和右子树在preorder和inorder里的范围,向左右边进行递归,构建左右子树。每一个递归由1开始继续做继续递归。

5。假设数组子集的范围由head和tail表示。一旦pre_head/in_head > pre_tail/in_tail(两者都可以),就表示该子树为空。返回null, 一旦pre_head/in_head == pre_tail/in_tail

表示的就是当前子树为单节点,返回new TreeNode(preorder[pre_start])即可。


给出代码:

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return helper(0, preorder.length - 1, 0, inorder.length - 1, preorder, inorder);
    }
    
    public TreeNode helper(int pre_start, int pre_end, int in_start, int in_end, int[] preorder, int[] inorder){
        if(in_start > in_end)
            return null;
        if(in_start == in_end)
            return new TreeNode(inorder[in_start]);
        int in_pos = 0;
        for(int i = in_start; i <= in_end; i++){
            if(preorder[pre_start] == inorder[i]){
                in_pos = i;
                break;
            }
        }
        TreeNode root = new TreeNode(preorder[pre_start]);
        root.left = helper(pre_start + 1, pre_start + in_pos - in_start, in_start, in_pos - 1,preorder, inorder);
        root.right = helper(pre_start + in_pos - in_start + 1, pre_end, in_pos + 1, in_end, preorder, inorder);
        return root;
    }


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值