这两个答案有什么区别？

Question

我完成了一个问题“平衡二叉树”，这是我编写的两个版本。第一个是接受的，另一个是失败的。不同的是“+1”的位置，当我把它放在return语句中时，算法通过了。

但是为什么我不能把它放在赋值语句中呢？

失败的测试用例是这个：

输入：[1,2,2,3,null,null,3,4,null,null,4] 输出：真 预期输出：假 （我不知道这个数组是怎么变成二叉树的。）

二叉树节点的定义。

public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode (int x) { val = x;}
}

以下是算法：

public boolean isBalanced(TreeNode root) {
    if(depth(root) == -1)
        return false;
    else return true;
}

// algorithm 1
private int depth(TreeNode root) {
    if(root == null)
        return 0;
    int leftdepth = depth(root.left);
    int rightdepth = depth(root.right);
    if(leftdepth == -1 || rightdepth == -1 || (Math.abs(leftdepth - rightdepth) > 1))
        return -1;
    else
        return Math.max(leftdepth,rightdepth) + 1;

// algorithm 2
private int depth(TreeNode root) {
    if(root == null)
        return 0;
    int leftdepth = depth(root.left)+1;
    int rightdepth = depth(root.right)+1;
    if(leftdepth == -1 || rightdepth == -1 || (Math.abs(leftdepth - rightdepth) > 1))
        return -1;
    else
        return Math.max(leftdepth,rightdepth);
}

Answer 1

因为您将这些值与 -1 进行比较。如果您愿意，第一个算法将与第二个算法相同：

if(leftdepth == 0 || rightdepth == 0 ...

Answer 2

算法 1 是正确的，因为分支的深度是其子分支的深度加上 1（此节点）的总和。

算法 2 不正确，因为它将null 节点的深度记录为1。