617 leetcode 合并二叉树

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

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
在这里插入图片描述

注意: 合并必须从两个树的根节点开始。
【树的结构】

 Definition for a binary tree node.
 struct TreeNode {
      int val;
      struct TreeNode *left;
      struct TreeNode *right;
  };

【题解】
一开始是这样想的,因为递归有点想不明白,所以想新建一个树,存起来新的树,再返回。一开始写的代码特别混乱没有逻辑……
之后改了一遍:

struct TreeNode* mergeTrees(struct TreeNode* t1, struct TreeNode* t2){
    if (t1 == NULL && t2 == NULL)
        return NULL;
    struct TreeNode *root; 
    root = malloc(sizeof(struct TreeNode));
    root->val = -1;
    root->left = NULL;
    root->right = NULL;
    if (t1 == NULL && t2 != NULL) {
        *root = *t2;
    }
    else if (t1 != NULL && t2 == NULL) {
        *root = *t1;
    }
    else if (t1 != NULL && t2 != NULL) {
        root->val = t1->val + t2->val;
        root->left = mergeTrees(t1->left, t2->left);
        root->right = mergeTrees(t1->right, t2->right);
    }
    return root;
}

【BUG及注意事项】
1.line 2 设置t1 t2的空判断是十分必要的,因为会存在之后递归到两个都为空的情况,这个时候如果不设置判断条件,会把指针初始化的内容录入(指针的内存分配在最前面),变成-1。
2.line 5-8 新树的初始化,不初始化会报错空指针。
3.line 10 line 13 当时看的题解是把val right left分别初始化,但是过于麻烦,进行简写,但是!没加*是不行的,不能写成root = t2。还有这一点很精明,如果一条线空而另外一条线不空,则不用再考虑不空线上有多少个节点,直接接在空线上即可。

【结果】
60ms 21.6MB很难看
速度:if 套的else if太多,相对于这种判断类型的语句执行速度,switch > if + if > if + elif
内存:可以通过直接修改形参之一,避免新建树增加内存。

再改:

struct TreeNode* mergeTrees(struct TreeNode* t1, struct TreeNode* t2){
    if (t1 == NULL && t2 == NULL)
        return NULL;
    if (t1 == NULL || t2 == NULL)
        return t1 == NULL ? t2 : t1;
    struct TreeNode *root; 
    root = malloc(sizeof(struct TreeNode));
    root->val = t1->val + t2->val;
    root->left = mergeTrees(t1->left, t2->left);
    root->right = mergeTrees(t1->right, t2->right);
    return root;
}

【BUG及注意事项】
1.line4 line 5通过使用 ’ ? ’ 来减少else if的使用,以求提升速度
2.将新树的分配空间过程后置,因为前面并不需要新树参与,并且省去了初始化过程。

【结果】
40ms 21.7MB
内存没有很大改观,如果想要降低内存还是要删除新树,直接更改一个形参

再改:

struct TreeNode* mergeTrees(struct TreeNode* t1, struct TreeNode* t2){
    if (t1 == NULL && t2 == NULL)
        return NULL;
    if (t1 == NULL && t2 != NULL) 
    	return t2;
    if (t1 != NULL && t2 == NULL) 
    	return t1;
    t1->val += t2->val;
    t1->left = mergeTrees(t1->left, t2->left);
    t1->right = mergeTrees(t1->right, t2->right);
    return t1;
}

【BUG及注意事项】
1.line 4-7删除了用 ’ ? ’ 的判断形势,换成了两个带返回的if,提高了速度,同时让程序可读性增强。
2.直接修改t1 ,而不是创建新树,从而降低内存。

【结果】

在这里插入图片描述
我好了嘿嘿嘿。

过程中借鉴了很多人的好的算法,来进一步改进代码,很感谢在评论区发代码的dalao们!进一步的修改没有找到更好的思路了,应该目前就是这样QWQ

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值