算法---LeetCode 617. 合并二叉树

1. 题目

原题链接

给定两个二叉树，想象当你将它们中的一个覆盖到另一个上时，两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠，那么将他们的值相加作为节点合并后的新值，否则不为 NULL 的节点将直接作为新二叉树的节点
。
示例 1:

输入:
Tree 1 Tree 2
1 2
/ \ / \
3 2 1 3
/ \ \
5 4 7
输出:
合并后的树:
3
/
4 5
/ \ \
5 4 7

注意: 合并必须从两个树的根节点开始。
Related Topics 树
👍 595 👎 0

2. 题解

2.1 解法1 : 递归 DFS 深度优先遍历

递归终止条件: t1 与 t1 都为空
合并根节点, 合并左子树,合并右子树
把函数看成合并子树

    class Solution {
        public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
            if (t1 == null) {
                return t2;
            }
            if (t2 == null) {
                return t1;
            }
            TreeNode merged = new TreeNode(t1.val + t2.val);
            merged.left = mergeTrees(t1.left, t2.left);
            merged.right = mergeTrees(t1.right, t2.right);
            return merged;
        }
    }

2.2 解法2: 迭代法(队列)

利用类层序遍历比较两棵树, 合并后的树保存在 t1 中

这里注意: 要先判断都不为空的情况, 若符合条件, 先将其加入队列, 如果放在后面判断, 由于出现了对 p1 赋值的1情况, 所以可能会导致原本不符合条件又变得符合条件了

    class Solution {
        public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
            if (t1 == null) {
                return t2;
            }
            if (t2 == null) {
                return t1;
            }

            Queue<TreeNode> queue = new LinkedList<>();
            queue.add(t1);
            queue.add(t2);
            while (!queue.isEmpty()) {
                TreeNode p1 = queue.poll();
                TreeNode p2 = queue.poll();
                p1.val += p2.val;

                // 这里注意, 要先判断都不为空的情况,
                // 若符合条件, 先将其加入队列
                // 如果放在后面判断, 由于出现了对 p1 赋值的1情况
                // 所以可能会导致原本不符合条件又变得符合条件了
                if (p1.left != null && p2.left != null) {
                    queue.add(p1.left);
                    queue.add(p2.left);
                }

                if (p1.right != null && p2.right != null) {
                    queue.add(p1.right);
                    queue.add(p2.right);
                }

                if (p1.left == null && p2.left != null) {
                    p1.left = p2.left;
                }
                if (p1.right == null && p2.right != null) {
                    p1.right = p2.right;
                }

            }
            return t1;


        }
    }

---

参考题解