秋招 - 算法 - 合并二叉树

76 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第26天,点击查看活动详情 >>

一、题目描述 LeetCode - 617

给你两棵二叉树: root1 和 root2 。
将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。返回合并后的二叉树。

示例 1:

image.png

输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出:[3,4,5,5,4,null,7]

示例 2:

输入:root1 = [1], root2 = [1,2]
输出:[2,2]

二、解题思路

本题要求将两个二叉树合并成一棵,两个树相同位置都有节点时将节点的值相加,一个树有而另一棵树没有时将不为空的节点作为新二叉树的节点。本题可以采用二叉树深度优先搜索来完成。对于在搜索过程中出现的每种情况采用不同的合并方式:

  • 当两个节点重叠时,将两节点的值相加作为新的二叉树的节点。
  • 当其中一个节点为null时,将不为null的节点作为新二叉树的节点。
  • 若两个节点均为null,则新二叉树此位置也为null。

三、代码

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

四、总结

本题使用深度优先搜索方法,通过遍历两棵二叉树来完成二叉树的合并。由于要递归遍历两个二叉树进行合并,所以时间复杂度为O(min(m,n))其中m,n为两棵二叉树的节点个数。空间复杂度为O(min(m,n))其中m,n分别表示两棵二叉数的层数。