java - 按深度加权的整数二叉树中的总和值

Question

这是我的参考树：

    3
   / \
  5   2
 /   / \
1   4   6

这是递归方法的预期输出：

(1*3) + (2 * (5 + 2)) + (3 * (1 + 4 + 6)) = 50

...这是我到目前为止的代码：

public int depthSum()
{
    int depth = 1;
    return depthSum(overallRoot, depth);
}

public int depthSum(IntTreeNode someNode, int someDepth)
{
    if(someNode == null)
    {
        return 0;
    }
    else
    {
        return someDepth * someNode.data + //some recursion
    }
}

我知道我可能不得不打电话给自己并增加一些深度，但我似乎无法做到这一点。有任何想法吗？

Answer 1

大概你的意思是：

return someDepth * someNode.data +
       depthSum(someNode.left, someDepth+1) +
       depthSum(someNode.right, someDepth+1);