中序/前序+中序/后序构建二叉树

中序/前序+中序/后序构建二叉树

根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。

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

    3
   / \
  9  20
    /  \
   15   7
class Solution {
    private Map<Integer, Integer> indexMap;

    public TreeNode myBuildTree(int[] preorder, int[] inorder, int preorder_left, int preorder_right, int inorder_left, int inorder_right) {
        if (preorder_left > preorder_right) {
            return null;
        }
        int preorder_root = preorder_left;
        int inorder_root = indexMap.get(preorder[preorder_root]);
        TreeNode root = new TreeNode(preorder[preorder_root]);
        int size_left_subtree = inorder_root - inorder_left;
        root.left = myBuildTree(preorder,inorder,preorder_left+1,preorder_left+size_left_subtree,inorder_left,inorder_root-1);
        root.right = myBuildTree(preorder, inorder, preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right);
        return root;
    }
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        int n = preorder.length;
        // 构造哈希映射,帮助我们快速定位根节点
        indexMap = new HashMap<Integer, Integer>();
        for (int i = 0; i < n; i++) {
            indexMap.put(inorder[i], i);
        }
        return myBuildTree(preorder, inorder, 0, n - 1, 0, n - 1);
    }
}

111

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

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

例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
    3
   / \
  9  20
    /  \
   15   7
class Solution {
    Map<Integer,Integer> map = new HashMap<Integer,Integer>();

    public TreeNode buildTree(int[] inorder, int[] postorder) {
        for(int i =0;i<inorder.length;i++){
            map.put(inorder[i],i);
        }
        return helper(inorder, postorder, postorder.length - 1, 0, inorder.length - 1);
    }
    //postEnd表示
    public TreeNode helper(int[] inorder, int[] postorder, int postEnd, int inStart, int inEnd) {
        if (inStart > inEnd) {
            return null;
        }
        //获取postorder里面的postEnd坐标的根节点,也就是最前面的根节点
        int currentVal = postorder[postEnd];
        //放入一个TreeNode中
        TreeNode current = new TreeNode(currentVal);
        //去存放inorder的hash表中找到这个下标方位inIndex作为分隔左右子树的
        int inIndex = map.get(current.val); 
        current.left = helper(inorder, postorder, postEnd - (inEnd- inIndex) - 1, inStart, inIndex - 1);
        current.right = helper(inorder, postorder, postEnd - 1, inIndex + 1, inEnd);
        return current;
    }
}

都是采用递归的思想,根据在中序遍历中根节点的位置决定往左遍历往右遍历的不同下标进行的操作。

展开阅读全文

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

©️2019 CSDN 皮肤主题: 游动-白 设计师: 上身试试
应支付0元
点击重新获取
扫码支付

支付成功即可阅读