leetcode 617 合并二叉树

简介: leetcode 617 合并二叉树

合并二叉树

2ecf2e839288400d94aa89c38892db6c.png

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* get_merge_tree(TreeNode* root1, TreeNode* root2)
    {
        if(root1 == nullptr && root2 == nullptr) return nullptr;
        else if(root1 != nullptr && root2 == nullptr) return root1;
        else if(root1 == nullptr && root2 != nullptr) return root2;
        TreeNode* new_root = new TreeNode;
    //合并树的值
        new_root->val = root1->val + root2->val;
        new_root->left = get_merge_tree(root1->left , root2->left);
        new_root->right = get_merge_tree(root1->right , root2->right);
        return new_root;
    }
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(root1 == nullptr && root2 == nullptr) return nullptr;
        else if(root1 != nullptr && root2 == nullptr) return root1;
        else if(root1 == nullptr && root2 != nullptr) return root2;
        return get_merge_tree(root1 ,root2);
    }
};

二刷

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* track_back(TreeNode* root1, TreeNode* root2)
    {
        if(root1 == nullptr && root2 == nullptr) return nullptr;
        else if(root1 != nullptr && root2 == nullptr) 
        {
            TreeNode* cur = new TreeNode(root1->val);
            cur->left = track_back(root1->left,nullptr);
            cur->right = track_back(root1->right,nullptr);
            return cur;
        }
        else if(root1 == nullptr && root2 != nullptr)
        {
           TreeNode* cur = new TreeNode(root2->val);
            cur->left = track_back(nullptr,root2->left);
            cur->right = track_back(nullptr,root2->right);
            return cur ;
        }else if(root1 != nullptr && root2 != nullptr)
        {
            TreeNode* cur = new TreeNode(root1->val + root2->val);
            cur->left = track_back(root1->left,root2->left);
            cur->right = track_back(root1->right,root2->right);
            return cur;
        }
        return nullptr;
    }
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        TreeNode* root = track_back(root1,root2);
        return root;
    }
};

相关文章
|
11天前
leetcode代码记录(二叉树的所有路径
leetcode代码记录(二叉树的所有路径
13 0
|
2天前
|
存储 Java
JAVA数据结构刷题 -- 力扣二叉树
JAVA数据结构刷题 -- 力扣二叉树
8 0
|
3天前
[LeetCode]—— 226——翻转二叉树
[LeetCode]—— 226——翻转二叉树
|
3天前
[LeetCode]——965——单值二叉树
[LeetCode]——965——单值二叉树
|
3天前
LeetCode——101——对称二叉树
LeetCode——101——对称二叉树
25 12
|
3天前
|
存储
LeetCode———144—— 二叉树的前序遍历
LeetCode———144—— 二叉树的前序遍历
|
3天前
LeetCode——965. 单值二叉树
LeetCode——965. 单值二叉树
|
4天前
|
存储 搜索推荐 C语言
Leetcode—合并两个有序数组—C语言
Leetcode—合并两个有序数组—C语言
|
5天前
力扣56.合并区间
力扣56.合并区间
|
5天前
|
算法
数据结构与算法⑮(第四章_下)二叉树OJ(力扣:144,965,104,110,226,100,101,572)(下)
数据结构与算法⑮(第四章_下)二叉树OJ(力扣:144,965,104,110,226,100,101,572)
7 1