617. 合并二叉树(经典递归)

2022-05-05 20:27:11 浏览数 (1)

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

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

思路:根据递归的思路 1.设置出口 2.操作 3.调用自身 那么我们可知,当当前节点t1为空,那么合并就是t2. 当当前节点t2为空,那么合并就是t1.

否则就执行操作,使得t1的值 t2的值(因为我们最终要返回t1作为新的合并的树)

然后就调用自身, 合并的左子树 = 函数(t1的左子树 ,t2左子树) 合并的右子树 = 函数(t1的右子树,t2的右子树)

代码语言:javascript复制
class Solution {
public:
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        if (t1 == NULL) return t2; // 如果t1为空,合并之后就应该是t2
        if (t2 == NULL) return t1; // 如果t2为空,合并之后就应该是t1
        // 修改了t1的数值和结构
        t1->val  = t2->val;                             // 中
        t1->left = mergeTrees(t1->left, t2->left);      // 左
        t1->right = mergeTrees(t1->right, t2->right);   // 右
        return t1;
    }
};

0 人点赞