坚持刷题｜合并二叉树

Hello，大家好，我是阿月。坚持话题，老年痴呆追不上我，今天刷：合并二叉树

题目

617.合并二叉树

破题点

• 理解二叉树的结构：二叉树由节点组成，每个节点有一个值以及左右子节点。
• 递归：知道每个结点要做的事情，采用递归方法合并两棵树。递归处理每个节点，对应位置上的节点值相加，左右子树也递归地合并。
• 边界条件处理：需要考虑两棵树中某些节点可能为空的情况，在递归过程中进行相应的判断和处理。
• 新节点的创建：在合并过程中，创建新的节点来存储合并后的结果，需要创建新的二叉树节点并设置合适的值。

代码实现

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

• 代码中使用了递归的方式来合并两棵二叉树。
• 首先，检查每个节点，如果两个树中相同位置的节点都存在，将它们的值相加，然后递归地合并左右子树。
• 如果某个节点在其中一个树中不存在，直接取另一棵树中对应位置的节点。