【打卡】牛客网：BM32 合并二叉树

最新推荐文章于 2024-09-03 21:49:34 发布

初霁i

最新推荐文章于 2024-09-03 21:49:34 发布

阅读量57

点赞数 1

文章标签：算法

本文链接：https://blog.csdn.net/weixin_47173826/article/details/134199614

版权

作者改进了合并两个二叉树的C++解决方案，移除了对空指针的单独检查，通过递归调用`mergeTrees`直接处理节点合并，简化了代码逻辑。

摘要由CSDN通过智能技术生成

自己写的：

算是这么久写的比较快的了。刚开始打算void merge并最终返回t1，但是发现处理t1==NULL的情况太麻烦了。

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param t1 TreeNode类
     * @param t2 TreeNode类
     * @return TreeNode类
     */
    TreeNode* merge(TreeNode* t1, TreeNode* t2) {
        if (t1 == NULL && t2 == NULL)
            return NULL;
        if (t2 == NULL)
            return t1;
        if (t1 == NULL)
            return t2;

        TreeNode* ans = new TreeNode (t1->val + t2->val);
        ans->left = merge(t1->left, t2->left);
        ans->right = merge(t1->right, t2->right);

        return ans;
    }
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        // write code here
        return merge(t1, t2);
    }
};

模板的：

根据模板修改了一下我的。

①不需要判断二者为空的情况，因为二者为空，返回另一个自然也是空。

②直接递归主函数。本人不太习惯这么写。

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param t1 TreeNode类
     * @param t2 TreeNode类
     * @return TreeNode类
     */

    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        // write code here
        if (t2 == NULL)
            return t1;
        if (t1 == NULL)
            return t2;

        TreeNode* ans = new TreeNode (t1->val + t2->val);
        ans->left = mergeTrees(t1->left, t2->left);
        ans->right = mergeTrees(t1->right, t2->right);

        return ans;
    }
};