我正在尝试实现该算法,但不断出现错误,我不太清楚为什么:
size = len(data)
total = sum(data)
barray = [[False for x in range(int(size+1))] for x in range(int((total/2)+1))]
for x in range(0,size):
barray[0][x] = True
for x in range(1,int(total/2)):
barray[x][0] = True
for i in range(1,int(total/2)):
for j in range(1,size):
barray[i,j] = (barray[i, j - 1] or barray[i - data[j - 1], j - 1]) if data[j-1] <= i else barray[i, j - 1]
return barray[int(total/2),size]
错误:
Traceback (most recent call last):
File "C:/Users/tuf21741/PycharmProjects/PyBackup/test.py", line 25, in <module>
assert checkio([10, 10]) == 0, "1st example"
File "C:/Users/tuf21741/PycharmProjects/PyBackup/test.py", line 17, in checkio
barray[i,j] = (barray[i, j - 1] or barray[i - data[j - 1], j - 1]) if data[j-1] <= i else barray[i, j - 1]
TypeError: list indices must be integers, not tuple
最佳答案
barray[i,j] = (barray[i, j - 1] or barray[i - data[j - 1], j - 1]) if data[j-1] <= i else barray[i, j - 1]
看起来你应该在这里使用单独的方括号作为索引。
barray[i][j] = (barray[i][j - 1] or barray[i - data[j - 1]][j - 1]) if data[j-1] <= i else barray[i][j - 1]
以及返回语句。
return barray[int(total/2)][size]