我需要在一个返回零值的列表中找到三个值。我的代码存在的问题是,它返回一个带有元组的嵌套列表,而不是列表。你知道吗
输入:
[-1, 0, 1, 2, -1, -4]
输出:
[(-1, -1, 2), (-1, 0, 1)]
预期产量:
[[-1, -1, 2], [-1, 0, 1]]
我的代码:
def threeSum(a):
N = len(a)
c=[]
for i in range(N):
for j in range(i+1, N):
for k in range(j+1, N):
if a[i]+a[j]+a[k] == 0:
c.append([a[i],a[j],a[k]])
b_set = set(tuple(sorted(x)) for x in c)
#remove duplicates
return list(b_set)
print(threeSum([-1, 0, 1, 2, -1, -4]))
插入此行
在您的预期输出返回语句之前
将您的退货声明更改为
相关问题 更多 >
编程相关推荐