算法提高之LeetCode刷题算法精习

Leetcode 617 合并二叉树

2020-01-09  本文已影响0人  小黑天天快乐

原文链接

题目

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

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

例子

思路

从根结点出发,对根结点进行合并,再分别递归合并其左右子树。

C++程序

TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
    if(t1==nullptr && t2==nullptr) return nullptr;
    if(t1==nullptr) return t2;
    if(t2==nullptr) return t1;
    else {
        t1->val += t2->val;
        t1->left = mergeTrees(t1->left, t2->left);
        t1->right = mergeTrees(t1->right, t2->right);
        return t1;
    }
}
上一篇下一篇

猜你喜欢

热点阅读