Hello,大家好,我是阿月。坚持话题,老年痴呆追不上我,今天刷:合并二叉树
题目
破题点
- 理解二叉树的结构:二叉树由节点组成,每个节点有一个值以及左右子节点。
- 递归:知道每个结点要做的事情,采用递归方法合并两棵树。递归处理每个节点,对应位置上的节点值相加,左右子树也递归地合并。
- 边界条件处理:需要考虑两棵树中某些节点可能为空的情况,在递归过程中进行相应的判断和处理。
- 新节点的创建:在合并过程中,创建新的节点来存储合并后的结果,需要创建新的二叉树节点并设置合适的值。
代码实现
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
public class MergeBinaryTrees {
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
if (t1 == null && t2 == null) {
return null;
}
int val = (t1 == null ? 0 : t1.val) + (t2 == null ? 0 : t2.val);
TreeNode newNode = new TreeNode(val);
// 递归处理左子树
newNode.left = mergeTrees(t1 == null ? null : t1.left, t2 == null ? null : t2.left);
// 递归处理右子树
newNode.right = mergeTrees(t1 == null ? null : t1.right, t2 == null ? null : t2.right);
return newNode;
}
public static void main(String[] args) {
// 示例用法
MergeBinaryTrees merger = new MergeBinaryTrees();
// 构建树1
TreeNode t1 = new TreeNode(1);
t1.left = new TreeNode(3);
t1.right = new TreeNode(2);
t1.left.left = new TreeNode(5);
// 构建树2
TreeNode t2 = new TreeNode(2);
t2.left = new TreeNode(1);
t2.right = new TreeNode(3);
t2.left.right = new TreeNode(4);
t2.right.right = new TreeNode(7);
// 合并树
TreeNode mergedTree = merger.mergeTrees(t1, t2);
// 示例输出:打印中序遍历合并后的树
printInorder(mergedTree);
}
public static void printInorder(TreeNode root) {
if (root != null) {
printInorder(root.left);
System.out.print(root.val + " ");
printInorder(root.right);
}
}
}
- 代码中使用了递归的方式来合并两棵二叉树。
- 首先,检查每个节点,如果两个树中相同位置的节点都存在,将它们的值相加,然后递归地合并左右子树。
- 如果某个节点在其中一个树中不存在,直接取另一棵树中对应位置的节点。