Leetcode刷题笔记 617. 合并二叉树

知识点：二叉树、递归
时间：2020年9月23日
题目链接：https://leetcode-cn.com/problems/merge-two-binary-trees/

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

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

示例1
输入：

		Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7       

输出：

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

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

解法

1. 可以用递归和迭代
2. 递归时注意返回的值

代码

#include <stdio.h>
#include <iostream>
#include <queue>
using namespace std;
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        if(t1==nullptr)return t2;
        if(t2==nullptr)return t1;
        t1->val += t2->val;
        t1->left = mergeTrees(t1->left,t2->left);
        t1->right = mergeTrees(t1->right,t2->right);
        return  t1;
    }
};
/*
class Solution {
public:
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        if(t1==nullptr) return t2;
        if(t2==nullptr) return t1;
        queue<TreeNode*> q;
        q.push(t1);
        q.push(t2);
        while(!q.empty()) {
            TreeNode* node1 = q.front(); q.pop();
            TreeNode* node2 = q.front(); q.pop();
            node1->val += node2->val;
            if (node1->left != nullptr && node2->left != nullptr) {
                q.push(node1->left);
                q.push(node2->left);
            }
            if (node1->right != nullptr && node2->right != nullptr) {
                q.push(node1->right);
                q.push(node2->right);
            }
            if (node1->left==nullptr && node2->left != nullptr) {
                node1->left=node2->left;
            }
            if (node1->right==nullptr && node2->right != nullptr) {
                node1->right=node2->right;
            }
        }
        return t1;
    }
};*/
void printt(TreeNode* node){
    if(node == nullptr)return;
    printt(node->left);
    cout<<node->val;
    printt(node->right);
}
int main()
{
    TreeNode node1(1);TreeNode node2(3);  node1.left = &node2;
    TreeNode node3(5);TreeNode node4(6);  node3.right = &node4;
    Solution s;
    TreeNode* root = s.mergeTrees(&node1, &node3);
    printt(root);
    return 0;
}

今天也是爱zz的一天哦！