合并2个二叉树-简单

Question

我正在尝试合并2棵二叉树，而不用担心使结果树平衡。这是我的解决方案，但不起作用。为什么Treenode ans和head从合并函数返回时设置为0。正如我所理解的，由于TreeNode不是原始类型，因此指向ans的head应该在调用合并函数后使用生成的树更新https://leetcode.com/problems/merge-two-binary-trees/

/**
 * 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) {
        
        
        
        TreeNode ans = new TreeNode();
        TreeNode head = ans;
        
        if(root1!=null && root2==null)
            return root1;
        else if (root1==null && root2!=null)
            return root2;
        else if (root1!=null &&root2!=null)
            merge (root1, root2,ans) ;   
            
            return head;
            
        
    }
    
    void merge (TreeNode root1, TreeNode root2,TreeNode ans)
    {
         if(root1!=null && root2==null)
            root2 = new TreeNode(0);
        
         else if (root1==null && root2!=null)
            root1 = new TreeNode(0);
         else if(root1==null &&root2==null)
         return;
        
            ans = new TreeNode(root1.val+root2.val); 
              merge(root1.left,root2.left,ans.left);
                merge(root1.right,root2.right,ans.right);          
    }
    
}

Answer 1

问题是变量head和ans不会像您期望的那样更新。Java是按值传递的，如这里所解释的。这意味着ans是一个引用，您将其值传递给函数合并，然后在那里覆盖。

您还可以通过每次递归调用merge重新分配ans的值。要返回正确的值，您需要在第一次调用时返回分配给该变量的值。

希望这有帮助。