leetcode-105 从前序与中序遍历序列构造二叉树(Medium)

题目描述

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

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

例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   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) {
        
    }
}

题解一

根据前序遍历,可以知道第一个元素是根节点。
根据中序遍历,可以获取根节点的下标,下标左边的子区间就是左子树,下标右边的子区间就是右子树。
在这里插入图片描述
这是一个递归的过程,每次确定前序遍历和中序遍历的范围,就能构造一个子树。
在中序遍历寻找根节点的过程可以使用Map做索引。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private int [] pre;
    private int [] in;
    private HashMap<Integer, Integer> map = new HashMap<>();

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        for (int i = 0 ; i < inorder.length ; i++) {
            map.put(inorder[i], i);
        }
        pre = preorder;
        in = inorder;
        return deep(0, preorder.length - 1, 0, preorder.length - 1);
    }



    private TreeNode deep(int start, int end, int start2, int end2)
    {
        if (start > end || start2 > end2) {
            return null;
        } else {
            TreeNode result = new TreeNode(pre[start]);
            int index = map.get(pre[start]);
            int leftLength = index - start2;
            int rightLength = end2 - index;
            if (leftLength > 0)
                result.left = deep(start + 1, start + 1 + leftLength, start2, index - 1);
            if (rightLength > 0)
                result.right = deep(start + leftLength + 1, end, index + 1, end2);
                    
            return result;
        }
    }
}

时间复杂度:O(n)O(n),其中 nn 是树中的节点个数。

空间复杂度:O(n)O(n),除去返回的答案需要的 O(n)O(n) 空间之外,我们还需要使用 O(n)O(n) 的空间存储哈希映射,以及 O(h)O(h)(其中 h 是树的高度)的空间表示递归时栈空间。这里 h<nh<n,所以总空间复杂度为O(n)O(n)

展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 技术黑板 设计师: CSDN官方博客
应支付0元
点击重新获取
扫码支付

支付成功即可阅读