错误：“int” object 在 python 中不可下标

Question

我正在尝试解决hackerearth中的问题，下面的代码给了我一个错误：

"int" object is not subscriptable- at line 4 (if(dp[g][p][d]:= -1):)

我知道当我们尝试索引 int obj 时会发生此错误，但这里 dp 是 3d 列表。

def max_val(i,g,p,d):
    if(i==n or g==maxG or p==maxP or d==maxD):
        return 0;
    if(dp[g][p][d] != -1):
        return dp[g][p][d]
    
    val=dp[g][p][d]
    if(g<maxG):
        val=max(val,gold[i]+max_val(i+1,g+1,p,d))
    if(p<maxP):
        val=max(val,plat[i]+max_val(i+1,g,p+1,d))
    if(d<maxD):
        val=max(val,dia[i]+max_val(i+1,g,p,d+1))

    return val;

n=int(input())

maxG,maxP,maxD=[int(x) for x in input().split()]

total=0

gold=[]
plat=[]
dia=[]
dp=[[-1 for x in range(n)] for x in range(n)]

for _ in range(n):
    x=[int(i) for i in input().split()]
    gold.append(x[0])
    plat.append(x[1])
    dia.append(x[2])

max_val(0,0,0,0)

Answer 1

dp=[[-1 for x in range(n)] for x in range(n)]将返回一个二维列表，因为您通过在 a 中执行[-1 for x in range(n)]来创建列表列表是外部的。 您可以创建一个如下所示的三维列表：

dp=[[[-1 for x in range(2)] for x in range(2)] for x in range(2)]
print(dp) # outputs [[[-1, -1], [-1, -1]], [[-1, -1], [-1, -1]]]
print(dp[0][0][0]) # outputs -1