我正在尝试编写一个程序,以查找一维数组中不相邻元素的最大和,到目前为止,我有这个:

def find_max_sum(arr):
    incl = 0
    excl = 0

    for i in arr:

        # Current max excluding i
        new_excl = excl if excl>incl else incl

        # Current max including i
        incl = excl + i
        excl = new_excl

    # return max of incl and excl
    return (excl if excl>incl else incl)


有用。我唯一的问题是,在不使用for循环的情况下,如何将该函数转换为递归函数?我的大脑似乎找不到办法。

最佳答案

第1步:重写函数,使其更具Pythonic

def find_max_sum(lst):
    incl, excl = 0, 0

    for i in lst:
        incl, excl = excl + i, max(excl, incl)

    return max(excl, incl)


步骤2:现在很容易将其重写为递归函数

def find_max_sum_recursive(lst, incl, excl):
    if len(lst) > 0:
        i = lst[0]
        incl, excl = excl + i, max(excl, incl)

        return find_max_sum_recursive(lst[1:], incl, excl)
    else:
        return incl, excl

def call_find_max_sum_recursive(lst):
    return max(find_max_sum_recursive(lst, 0, 0))


现在你打电话

>>> call_find_max_sum_recursive([1, 2, 10, 3, 4])
15


步骤1并非绝对必要。毕竟,这涉及到for i in arr:中的代码块以及它们如何影响inclexcl。但是,这可能会帮助您重写。

08-24 21:13