Keshawn_lu's Blog

Leetcode 617. 合并二叉树

字数统计: 317阅读时长: 1 min
2020/09/23 Share

题目简介:

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

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

示例 1:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
输入: 
Tree 1 Tree 2
1 2
/ \ / \
3 2 1 3
/ \ \
5 4 7
输出:
合并后的树:
3
/ \
4 5
/ \ \
5 4 7

注意: 合并必须从两个树的根节点开始。

思路:

常规深搜即可。

若两个结点都为空,则直接返回空;若两个结点中有一个为空,则返回另一个;若两个都不为空,则建立一个新结点,其val值为两个结点的和。

代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {

if(!t1 && !t2)
return NULL;

if(!t1)
return t2;
if(!t2)
return t1;

TreeNode* root = new TreeNode(t1 -> val + t2 -> val);

root -> left = mergeTrees(t1 -> left, t2 -> left);
root -> right = mergeTrees(t1 -> right, t2 -> right);

return root;
}
};
CATALOG
  1. 1. 题目简介:
  2. 2. 思路:
  3. 3. 代码如下: