zlatan的算法笔记16--合并二叉树（力扣617题）

最新推荐文章于 2024-08-14 12:02:23 发布

zlatanYang

最新推荐文章于 2024-08-14 12:02:23 发布

阅读量160

点赞数

本文链接：https://blog.csdn.net/qq_43408360/article/details/127851569

版权

算法笔记

本题对于二叉树进行学习，主要特点在于同时对于两棵二叉树进行操作，习惯操作后本题就很容易解决。

题目描述

在这里插入图片描述

解题步骤

在选择参数方面，传入两颗二叉树的根节点就可以了，返回值为合并后的二叉树根节点。
在终止条件处，如果第一棵树节点为空，就返回第二棵树，合并后就是第二棵树；同理对于另一棵树来说也是一样的，至此就可以确定终止条件。
当两个树节点都存在时，就可以将两棵树的节点值相加就合并了，这里既可以新建一个root3去装合并的结果，也可以省空间，直接覆盖在某一棵树上面就可以，本题我使用的是第二种方法。对于本道题三种遍历方式都可以，可以按照自己的喜欢去选择，本题使用的遍历方法为前序遍历。

代码部分

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(root1 == nullptr) {
            return root2;
        }
        if(root2 == nullptr) {
            return root1;
        }
        if(root1 != nullptr && root2 != nullptr) {
            root1->val = root1->val + root2->val;
        }
        root1->left = mergeTrees(root1->left, root2->left);
        root1->right = mergeTrees(root1->right, root2->right);
        return root1;
    }
};