在python中找到树的最大总和

时间:2014-09-20 21:25:58

标签: python tree

我有一个数字树,我希望能够找到数字的总和。在每个数字下面是左右两个孩子在所有可能的路径中,我希望能够通过所有可能的路径找到最大的数字。这是一个例子

       8
   3      11
10   2  32  6

返回8 + 11 + 32 = 51

我觉得这是一个递归问题,但我遇到了我的代码,并且一直遇到错误。我认为我接近这个错误。以下是我的代码:

# Returns root key value
def getRootValue(root):
    return root

# Returns reference to left child
def getLeftChild(root):
    value=None
    if root.leftChild!=None:
        value=root.leftChild
    return value

# Returns reference to right child
def getRightChild(root):
    value=None
    if root.rightChild!=None:
        value = root.rightChild
    return value

def sum_of_branch(root):
    sum=0  
if root.getLeftChild() ==None and root.getRightChild()==None:
    return rounds
else:
    rounds+=rounds+1
    keys_sum[depth]=sum+root.key
    return sum_to_deepest(root.left), sum_to_deepest(root.right)
    if root.getLeftChild()!=None:
        rounds+=root.getLeftChild().branchLenSum()
    if root.getRightChild()!=None:
        rounds+=root.getRightChild().branchLenSum()
    return rounds

1 个答案:

答案 0 :(得分:2)

在不知道您使用的数据结构的情况下很难给出答案。但我认为你正在寻找像这样的东西:

def sum_of_branch(root):
    # If it has not childs we have arrived at the end of the tree.
    # We return the value of this child.
    if root.getLeftChild() ==None and root.getRightChild()==None:
        return getRootValue(root)
    else:
        # If it has children we calculate the sum of each branch. 
        leftSum = sum_of_branch(root.getLeftChild())
        rightSum = sum_of_branch(root.getRightChild())
        # And return the maximun of them.
        if leftSum > rightSum:
            return getRootValue(root) + leftSum
        else:
            return getRootValue(root) + rightSum