构造二叉树

106 从中序和后序遍历序列构造二叉树

给定两个整数数组 inorder 和 postorder ,其中 inorder 是二叉树的中序遍历, postorder 是同一棵树的后序遍历,请你构造并返回这颗 二叉树 。

思路:后续遍历最后一个值为根节点,利用hashMap存储中序遍历,方便后续找到中序遍历的下标。利用四个指针,分别指向中序遍历和后序遍历的下标位置。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    HashMap<Integer,Integer> memo = new HashMap();
    int[] post;
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        for(int i=0;i<inorder.length;i++){
            memo.put(inorder[i],i);
        }
        post = postorder;
        TreeNode root = buildTree(0,inorder.length-1,0,post.length-1);
        return root;
    }
    public TreeNode buildTree(int inLeft,int inRight,int poLeft,int poRight){
        if(inLeft>inRight||poLeft>poRight) return null;
        int root = post[poRight];
        int inIndex = memo.get(root);
        TreeNode node = new TreeNode(root);
        node.left = buildTree(inLeft,inIndex-1,poLeft,poLeft+inIndex-1-inLeft);
        node.right = buildTree(inIndex+1,inRight,poLeft+inIndex-inLeft,poRight-1);
        return node;
    }
}

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

给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。

思路:与上题差不多,前序遍历序列开头为中心元素。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    HashMap<Integer,Integer> memo = new HashMap();
    int[] pre;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        for(int i=0;i<inorder.length;i++){
            memo.put(inorder[i],i);
        }
        pre = preorder;
        TreeNode root = buildTree(0,inorder.length-1,0,preorder.length-1);
        return root;
    }
    public TreeNode buildTree(int inLeft,int inRight,int preLeft,int preRight){
        if(inLeft>inRight||preLeft>preRight) return null;
        int root = pre[preLeft];
        int inIndex = memo.get(root);
        TreeNode node = new TreeNode(root);
        node.left = buildTree(inLeft,inIndex-1,preLeft+1,preLeft+inIndex-inLeft);
        node.right = buildTree(inIndex+1,inRight,preLeft+inIndex-inLeft+1,preRight);
        return node;
    }
}

19 最大二叉树

给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:

创建一个根节点,其值为 nums 中的最大值。
递归地在最大值 左边 的 子数组前缀上 构建左子树。
递归地在最大值 右边 的 子数组后缀上 构建右子树。
返回 nums 构建的 最大二叉树 。

思路:定义一个函数,可以寻找自定义始末位置的数组中的最大值。递归函数首先调用该函数,寻找到最大值,再从hash表中拿出下标记,同时递归求左右子树,更新范围即可。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    int[] numsTmp;
    HashMap<Integer,Integer> memo = new HashMap();
    public TreeNode constructMaximumBinaryTree(int[] nums) {
        for(int i=0;i<nums.length;i++){
            memo.put(nums[i],i);
        }
        numsTmp = nums;
        TreeNode root = constructMaximumBinaryTree(0,nums.length-1);
        return root;
    }
    public TreeNode constructMaximumBinaryTree(int start,int end){
        if(start>end) return null;
        int root = getMax(numsTmp,start,end);
        int inIndex = memo.get(root);
        TreeNode node = new TreeNode(root);
        node.left = constructMaximumBinaryTree(start,inIndex-1);
        node.right = constructMaximumBinaryTree(inIndex+1,end);
        return node;
    }
    public int getMax(int[] nums,int start,int end){
        int max = nums[start];
        for(int i=start;i<=end;i++){
            if(nums[i]>=max){
                max = nums[i];
            }
        }
        return max;
    }
}
  • 4
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值