动态规划算法例题

题目:
有一个序列[3,-5, 9,0,-2,8,-3,9,5,-9,-2,8],取其中不相邻子序列,求可能的最大值是多少?

思路:
用动态规划来求解。
对于最后一个元素,可以分成两种情况,取还是不取。如果取,那么我们在8的基础上再加上前10个元素组成的序列求得的最大值,如果不取,那么我们考虑前11个元素组成的序列即可。假设opt(i)表示(序列长度为i)根据题意所求的最大值。
则有,opt(12) = max(opt(10)+8, opt(11))
将式子一般化可得,opt(i) = max(opt(i-2)+8, opt(i-1))

下面用递归和迭代两种方法用代码实现:

import numpy as np
arr =[3,-5, 9,0,-2,8,-3,9,5,-9,-2,8def rec_opt(arr,i):
    if i==0:
        return arr[0]
    elif i==1:
        return max(arr[0],arr[1])
    else:
        A = rec_opt(i-2)+arr[i]
        B = rec_opt(i-1)
        return max(A,B)
def dp_opt(arr):
    opt = np.zeros(len(arr))
    opt[0] = arr[0]
    opt[1] = max(arr[0],arr[1])
    for i in range(2,len(arr)):
        A = opt[i-2]+arr[i]
        B = opt[i-1]
        opt[i] = max(A,B)
    return opt[-1]  

猜你喜欢

转载自blog.csdn.net/weixin_42936560/article/details/81812190