【Leetcode 617】Merge Two Binary Trees -EASY

题目

Given two binary trees and imagine that when you put one of them to cover the other, some nodes of the two trees are overlapped while the others are not.

You need to merge them into a new binary tree. The merge rule is that if two nodes overlap, then sum node values up as the new value of the merged node. Otherwise, the NOT null node will be used as the node of new tree.

给出两个二叉树，当你用其中一棵二叉树覆盖另外一棵树时，一些结点会重叠，一些不会。
你需要将这两棵树合并成一棵树。合并规则是如果两颗树的对应结点重合，则将两结点的值的和作为新结点的值。如果不重合，则非空结点将被作为新树的新结点。

---

示例：

---

Notes:The merging process must start from the root nodes of both trees.

思路

该题可以采用递归的方式。

题解

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
        if(t1 == null)
            return t2;
        if(t2 == null)
            return t1;
        t1.val += t2.val;
        t1.left = mergeTrees(t1.left,t2.left);
        t1.right = mergeTrees(t1.right,t2.right);
        return t1;
    }
}

反思

复杂度分析

时间复杂度：O（n）
空间复杂度：
O（n）(最差情形下)
O（log n）(平均情形下)

思路反思

太久没有做算法题，大脑一片混沌。递归呀！最简单的递归呀！！竟然没有第一时间想出来。

扩展

考虑不使用递归来遍历二叉树。

如何遍历呢？我们开辟一块栈空间用来存放结点对。

首先将两棵二叉树的根结点对放到栈stack中。

接下来是一个循环，每次循环从栈中取出一个结点对，如果两结点任一为空，则continue；否则，将两结点的值相加的结果赋给结点1。（因为二叉树1是我们最终想要的新的二叉树。）如果结点1的左子结点或右子结点为空，则将结点2的左子节点或右子结点赋给结点1，否则，则将该结点对加到stack中。

循环结束的条件是栈为空，此时所有的结点对都被访问到并被处理。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
        if(t1 == null)
            return t2;
        Stack <TreeNode[]> stack = new Stack<>();
        stack.push(new TreeNode[]{t1,t2});
        while(!stack.isEmpty()){
            TreeNode[] t = stack.pop();
            if(t[0]==null||t[1]==null){
                continue;
            }
            t[0].val+=t[1].val;
            if(t[0].left == null){
                t[0].left = t[1].left;
            }else{
                stack.push(new TreeNode[]{t[0].left,t[1].left});
            }
            if(t[0].right == null){
                t[0].right = t[1].right;
            }else{
                stack.push(new TreeNode[]{t[0].right,t[1].right});
            }
        }
        return t1;
    }
}

时间复杂度：O（n）
空间复杂度：O（n）

使用额外的栈空间来对二叉树进行遍历的方法是我第一次遇到的，对我的思路是一个扩展，很值得学习。