[LeetCode] 617.合并二叉树

LeetCode 617.合并二叉树

---

思路

递归

仍然采用递归的“三步法”：

• 1）确定递归函数的参数和返回值

  传入参数为两个二叉树的根节点

  返回值为合并之后二叉树的根节点

• 2）确定终止条件

  当一个树的节点为空时，则返回另一个树的节点

• 3）确定单层递归逻辑

  假设两棵树分别为tree1、tree2，通过重复利用tree1，将合并后的元素值重新赋给tree1。tree1的左子树是：合并 tree1左子树 tree2左子树之后的左子树。tree1的右子树：是 合并 tree1右子树 tree2右子树之后的右子树。最终返回tree1，就是合并后的根节点。

代码

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;
    }
};

迭代

使用队列，模拟层序遍历

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        if (t1 == NULL) return t2;
        if (t2 == NULL) return t1;
        queue<TreeNode*> que;
        que.push(t1);
        que.push(t2);
        while(!que.empty()) {
            TreeNode* node1 = que.front(); que.pop();
            TreeNode* node2 = que.front(); que.pop();
            // 此时两个节点一定不为空，val相加
            node1->val += node2->val;

            // 如果两棵树左节点都不为空，加入队列
            if (node1->left != NULL && node2->left != NULL) {
                que.push(node1->left);
                que.push(node2->left);
            }
            // 如果两棵树右节点都不为空，加入队列
            if (node1->right != NULL && node2->right != NULL) {
                que.push(node1->right);
                que.push(node2->right);
            }

            // 当t1的左节点 为空 t2左节点不为空，就赋值过去
            if (node1->left == NULL && node2->left != NULL) {
                node1->left = node2->left;
            }
            // 当t1的右节点 为空 t2右节点不为空，就赋值过去
            if (node1->right == NULL && node2->right != NULL) {
                node1->right = node2->right;
            }
        }
        return t1;
    }
};