代码随想录day17 Java版 二叉树部分

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

手算很容易，实现很困难的题目。思路是用后序确定根节点，再用中序切分。

具体切分过程比较像快速排序的左右区间切分（因为中序序列有根节点切分左右子树），也很自然地用到递归。那么递归结束条件为区间扫描完毕，然后找到后序遍历的最后一个元素在中序遍历中的位置,保存中序左子树个数，用来确定后序数列的个数。

class Solution {
    Map map= new HashMap<>();
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        for (int i = 0; i < inorder.length; i++) map.put(inorder[i],i);
        return findNode(inorder,0,inorder.length,postorder,0,postorder.length);
    }
    public TreeNode findNode(int[] inorder, int inBegin, int inEnd, int[] postorder, int postBegin, int postEnd) {
        if (inBegin >= inEnd || postBegin >= postEnd) return null;
        TreeNode node = new TreeNode(inorder[map.get(postorder[postEnd - 1])]);
        node.left = findNode(inorder, inBegin, map.get(postorder[postEnd - 1]),postorder, postBegin, postBegin + map.get(postorder[postEnd - 1])- inBegin);
        node.right = findNode(inorder, map.get(postorder[postEnd - 1]) + 1, inEnd,postorder, postBegin + map.get(postorder[postEnd - 1])- inBegin, postEnd - 1);
        return node;
    }
}

654.最大二叉树

题目说输入的数组大小一定是大于等于1的，所以我们不用考虑小于1的情况，那么当递归遍历的时候，如果传入的数组大小为1，说明遍历到了叶子节点了。那么应该定义一个新的节点，并把这个数组的数值赋给新的节点，然后返回这个节点。 这表示一个数组大小是1的时候，构造了一个新的节点，并返回。

class Solution {
    public TreeNode constructMaximumBinaryTree(int[] nums) {
        return constructMaximumBinaryTree1(nums, 0, nums.length);
    }
    public TreeNode constructMaximumBinaryTree1(int[] nums, int leftIndex, int rightIndex) {
        if (rightIndex - leftIndex <= 0) return null;
        if (rightIndex - leftIndex == 1) return new TreeNode(nums[leftIndex]);
        int maxIndex = leftIndex;
        int maxVal = nums[maxIndex];
        for (int i = leftIndex + 1; i < rightIndex; i++) {
            if (nums[i] > maxVal){
                maxVal = nums[i];
                maxIndex = i;
            }
        }
        TreeNode node = new TreeNode(maxVal);
        node.left = constructMaximumBinaryTree1(nums, leftIndex, maxIndex);
        node.right = constructMaximumBinaryTree1(nums, maxIndex + 1, rightIndex);
        return node;
    }
}

617.合并二叉树

比较简单，直接前序遍历

class Solution {
    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        if (root1 == null) return root2;
        if (root2 == null) return root1;
        root1.val += root2.val;
        root1.left = mergeTrees(root1.left,root2.left);
        root1.right = mergeTrees(root1.right,root2.right);
        return root1;
    }
}