【Leetcode】617. 合并二叉树

最新推荐文章于 2022-01-24 22:42:38 发布

J1JJOSH

最新推荐文章于 2022-01-24 22:42:38 发布

阅读量175

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： Leetcode 617 easy

本文链接：https://blog.youkuaiyun.com/zzy0006/article/details/89402567

Leetcode 专栏收录该内容

86 篇文章

订阅专栏

本文详细解析了一种用于合并两个二叉树的算法，通过递归方式实现节点值的相加，适用于节点重叠的情况。文章提供了两种实现方法，包括详细的代码示例，帮助读者理解并掌握该算法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

QUESTION

easy

题目描述

给定两个二叉树，想象当你将它们中的一个覆盖到另一个上时，两个二叉树的一些节点便会重叠。

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠，那么将他们的值相加作为节点合并后的新值，否则不为 NULL 的节点将直接作为新二叉树的节点。

示例 1:

输入:
    Tree 1                     Tree 2
          1                         2
         / \                       / \
        3   2                     1   3
       /                           \   \
      5                             4   7
输出:

合并后的树:
         3
        / \
       4   5
      / \   \
     5   4   7

说明

合并必须从两个树的根节点开始。

SOLUTION

递归很简单。

方法一

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        if(t1 == nullptr && t2 == nullptr) return nullptr;
        TreeNode* node = new TreeNode(0);
        if(t1 != nullptr && t2 == nullptr){
            node->val += t1->val;
            node->left = mergeTrees(t1->left, nullptr);
            node->right = mergeTrees(t1->right, nullptr);
        }
        else if(t1 == nullptr && t2 != nullptr){
            node->val += t2->val;
            node->left = mergeTrees(nullptr, t2->left);
            node->right = mergeTrees(nullptr, t2->right);
        }
        else{
            node->val += t1->val + t2->val;
            node->left = mergeTrees(t1->left, t2->left);
            node->right = mergeTrees(t1->right, t2->right);
        }
        return node;
    }
};

方法二

可以把代码写的稍微简洁点。

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        if(t1 == nullptr && t2 == nullptr) return nullptr;
        TreeNode* node = new TreeNode(0);
        if(t1 != nullptr) node->val += t1->val;
        if(t2 != nullptr) node->val += t2->val;
        node->left = mergeTrees((t1 == nullptr ? t1 : t1->left), (t2 == nullptr ? t2 : t2->left));
        node->right = mergeTrees((t1 == nullptr ? t1 : t1->right), (t2 == nullptr ? t2 : t2->right));
        return node;
    }
};