算法现在就是大厂、外企的硬指标。开发、测开、测试,想往上总是绕不开的。
题目描述
难度:【简单】 标签:【二叉树】
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。 你需要将他们合并为一个新的二叉树。 合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值, 否则不为 NULL 的节点将直接作为新二叉树的节点。
题目地址:https://leetcode-cn.com/problems/merge-two-binary-trees/description/
示例
输入: Tree 1 Tree 2 1 2 / \ / \ 3 2 1 3 / \ \ 5 4 7 输出: 合并后的树: 3 / \ 4 5 / \ \ 5 4 7
注意: 合并必须从两个树的根节点开始。
解题
合并必须从两个树的根节点开始。按照套路来,首先想好当前节点要做的事情,然后剩下的事情交给递归,不要陷入递归细节。
那么当前传入的 2 个节点,就是相互比较,有 3 种情况:
root1 == null && root2 == null
,都为空 null,直接返回 nullroot1 == null || root2 == null
,其中一个为 null,返回另一个不为空的节点- 剩下就是2个都不为空,那么就把两个值相加,赋值给新的节点
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode() {} * TreeNode(int val) { this.val = val; } * TreeNode(int val, TreeNode left, TreeNode right) { * this.val = val; * this.left = left; * this.right = right; * } * } */ class Solution { public TreeNode mergeTrees(TreeNode root1, TreeNode root2) { if (root1 == null && root2 == null) { return null; } if (root1 == null) { return root2; } if (root2 == null) { return root1; } // 都不为空,就进行相加,创建一个新的节点 TreeNode newNode = new TreeNode(root1.val + root2.val); // 前序遍历框架部分,分布递归2个节点的左孩子和右孩子 newNode.left = mergeTrees(root1.left, root2.left); newNode.right = mergeTrees(root1.right, root2.right); return newNode; } }