python挖矿_动态规划 python 挖矿问题

#!/usr/bin/env python

# encoding: utf-8

import copy

def good(n,w,g=[],p=[]):

# n为金矿数，w为人数，g为金矿数组，p为人数数组

arr = [0]*w

for i in range(w):

if (i+1)>=p[0]: # i为坐标， i+1为人数

arr[i] = g[0]

res = copy.deepcopy(arr) #深copy

print(res)

# 上面为只有一个金矿的情况

for i in range(1,n): # 金矿数

# 每一层代表前N个金矿人工数(1~10)人的解

# 前一层已经求出相同人数时选取不同金矿的最大值所以可以叠加

for j in range(w): # 人工数

if (j+1)

arr[j] = res[j] #和上一次情况相同

else:

t = 0 if (j-p[i])<0 else j-p[i] #防止负数取到后面的值

arr[j] = max(res[j],res[t]+g[i]) #挖和不挖第i座金矿比较取大

#res[t]+g[i] 为 挖第i座金矿的情况，res[已有人数 - 第i座所需人数]+第i座金子g[i]

res = copy.deepcopy(arr)

print(res)

return res.pop()

if __name__ == '__main__':

res = good(5,10,[400,500,200,300,350],[5,5,3,4,3])