地址 https://leetcode-cn.com/problems/merge-two-binary-trees/
给你两棵二叉树: root1 和 root2 。 想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。 你需要将这两棵树合并成一棵新二叉树。 合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值; 否则,不为 null 的节点将直接作为新二叉树的节点。 返回合并后的二叉树。 注意: 合并过程必须从两个树的根节点开始。 示例 1:
输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7] 输出:[3,4,5,5,4,null,7] 示例 2: 输入:root1 = [1], root2 = [1,2] 输出:[2,2] 提示: 两棵树中的节点数目在范围 [0, 2000] 内 -10^4 <= Node.val <= 10^4
解法
/** * 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: void dfs(TreeNode*& r1, TreeNode*& r2){ if(r2==NULL) return; if(r1==NULL&&r2!=NULL){ r1=r2;return; } r1->val += r2->val; dfs(r1->left,r2->left); dfs(r1->right,r2->right); } TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) { dfs(root1,root2); return root1; } };
我的视频题解空间