这道题需要操控两棵二叉树,并且是"同步"遍历,如果两个二叉树的节点都不为空,就可以直接相加,这一步作为递归函数处理逻辑,如果有一方为空,就可以退出递归,这是递归终止条件。为了便于思考,我们采用前序遍历,也就是从根节点往下走,每遍历到一个节点执行处理逻辑。大家可以结合我下面的代码以及注释理解此题。
代码及注释如下:
/**
* 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) {
//终止条件
if(root1 == NULL) return root2;
if(root2 == NULL) return root1;
//处理逻辑
root1 -> val += root2 -> val;
//递归左子树
root1 -> left = mergeTrees(root1 -> left,root2 -> left);
//递归右子树
root1 -> right = mergeTrees(root1 -> right,root2 -> right);
return root1;
}
};