天天看点

动态规划-挖矿问题

有一个国家发现了W座金矿,每座金矿的黄金储量不同,需要参与挖掘的工人数也不同。参与挖矿工人的总数是P人.每座金矿要么全挖,要么不挖,不能派出一半人挖取一半金矿。要求用程序求解出,要想得到尽可能多的黄金,应该选择挖取哪几座金矿?

#个人定义:m:金矿座数    n:工人总数    p[]:每座矿所需要的工人数    c[]:每座矿的的黄金数
m,n=5,10
def mining(p=[]*m,c=[]*m):
    preresult= [0] * n
    result = [0] * n
    for i in range(n):
        if i+1<p[0]:
            result[i]=0
        else:
            result[i]=c[0]

    for i in range(m):
        for j in range(n):
            if j+1<p[i]:
                result[j]=preresult[j]
            else:
                result[j]=max(preresult[j], preresult[j-p[i]+1] + c[i])

        preresult= result
    # print result
    return result[n-1]

p1=[5,5,3,4,3]
c1=[400,500,200,300,350]
print (mining(p1,c1))

p1=c1
print max(1,1+2)