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

剑指Offer:面试题07.重建二叉树

时间:2020-06-26 16:49:59      阅读:54      评论:0      收藏:0      [点我收藏+]

标签:dex   查找   order   new   ref   假设   range   hashmap   建二叉树   

1.要点

  • 二叉树遍历
  • 重复在查找中查找某个数时(重复+查找=二重循环),应考虑是否可以用map

2.题目

输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

3.示例

给出 前序遍历 preorder =?[3,9,20,15,7] 中序遍历 inorder = [9,3,15,20,7] 返回如下的二叉树:
    3
   /   9  20
    /     15   7

4.限制:

0 <= 节点个数 <= 5000

5.代码

public TreeNode buildTree(int[] preorder, int[] inorder) {
        return preorder.length == 0 ? null : buildNode(preorder, inorder);
    }

    public TreeNode buildNode(int[] preorder, int[] inorder) {
        int rootNodeIndex = -1;
        int rootNodeValue = 0;
        Map<Integer,Integer> inorderMap = new HashMap<>();
        for (int i = 0; i < inorder.length; i++) {
            inorderMap.put(inorder[i],i);
        }

        for (int i = 0; i < preorder.length; i++) {
                if (null != inorderMap.get(preorder[i])) {
                    rootNodeIndex = inorderMap.get(preorder[i]);
                    rootNodeValue = preorder[i];
                    break;
                }
        }

        TreeNode treeNode = new TreeNode(rootNodeValue);
        treeNode.left = rootNodeIndex > 0 ?
                buildNode(preorder, Arrays.copyOfRange(inorder, 0, rootNodeIndex)) :
                null;
        treeNode.right = rootNodeIndex < inorder.length - 1 ?
                buildNode(preorder, Arrays.copyOfRange(inorder, rootNodeIndex + 1, inorder.length)) :
                null;
        return treeNode;
    }


查看原文:http://yuyy.info/%e7%ae%97%e6%b3%95%e8%ae%ad%e7%bb%83/%e5%89%91%e6%8c%87offer%e9%9d%a2%e8%af%95%e9%a2%9807-%e9%87%8d%e5%bb%ba%e4%ba%8c%e5%8f%89%e6%a0%91/

剑指Offer:面试题07.重建二叉树

标签:dex   查找   order   new   ref   假设   range   hashmap   建二叉树   

原文地址:https://www.cnblogs.com/yuyy114/p/13195016.html

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