1. 程式人生 > 其它 >leetcode-617. 合併二叉樹

leetcode-617. 合併二叉樹

/**
 * 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* mergeTrees(TreeNode* root1, TreeNode* root2) { TreeNode* root = NULL; root = merge(root1,root2); return root; } TreeNode* merge(TreeNode* root1, TreeNode* root2){ int res = 0; TreeNode* root = NULL; // 建立跟節點 if
(root1==NULL&&root2==NULL) return NULL; if(root1==NULL){ res = root2->val; root = new TreeNode(res); root->left = merge(NULL,root2->left); root->right = merge(NULL,root2->right); } if(root2==NULL){ res
= root1->val; root = new TreeNode(res); root->left = merge(root1->left,NULL); root->right = merge(root1->right,NULL); } if(root1!=NULL&&root2!=NULL){ res = root1->val + root2->val; root = new TreeNode(res); root->left = merge(root1->left,root2->left); root->right = merge(root1->right,root2->right); } return root; } };
class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(root1==NULL)
            return root2;
        if(root2==NULL)
            return root1;
        TreeNode* root = new TreeNode(root1->val + root2->val);
        root->left = mergeTrees(root1->left, root2->left);
        root->right = mergeTrees(root1->right, root2->right);

        return root;
    }

};