• [LeetCode] 617. Merge Two Binary Trees


    Given two binary trees and imagine that when you put one of them to cover the other, some nodes of the two trees are overlapped while the others are not.

    You need to merge them into a new binary tree. The merge rule is that if two nodes overlap, then sum node values up as the new value of the merged node. Otherwise, the NOT null node will be used as the node of new tree.

    Example 1:

    Input:
    	Tree 1                     Tree 2                  
              1                         2                             
             /                        /                             
            3   2                     1   3                        
           /                                                    
          5                             4   7                  
    Output: 
    Merged tree:
    	     3
    	    / 
    	   4   5
    	  /     
    	 5   4   7
    

    Note: The merging process must start from the root nodes of both trees.

    合并二叉树,看到题就想到了递归版本的解法

    整个过程分为
    1.合并根节点
    2.合并左子树
    3.合并右子树

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

    再看看LeetCode的其他的写法,直接使用传进来的t1指针存储生成的树,生成的树中的节点也是从t1t2中得到的,省下新建TreeNode的时间

    从LeetCode上的提交来看,递归就是这个问题的最优解了,这是我没想到的

    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2)
    {
        if (!t1) return t2;
        if (!t2) return t1;
    
        t1->val += t2->val;
        t1->left = mergeTrees(t1->left, t2->left);
        t1->right = mergeTrees(t1->right, t2->right);
    
        return t1;
    }
    
  • 相关阅读:
    OO第四次暨期末总结
    OO第九到十一次作业小结
    OO第五到七次作业小结
    OO前三次作业阶段小结
    数据预处理相关
    Latex+VScode安装
    python学习网站+查询网站
    arcgis画图中添加带箭头的直线
    在ArcGIS 中标注中竖排文字
    vs2015使用fopen时遇到unsafe问题
  • 原文地址:https://www.cnblogs.com/arcsinw/p/9409770.html
Copyright © 2020-2023  润新知