题目:
给定一个数列,求的问题可以转换为求:, 我们将输入的list排序后,给定nums[i], 我们需要寻找 nums[j]+nums[k] = - nums[i], 这里的 。
题目要求不可包含重复的解,对于排序后的数组,如果num[i] = num[i+1],那么他们获得的nums[j]和nums[k]肯定是相同的(如果解存在的话),所以应该跳过这种情况。同理,如果num[j] = num[j+1], 那么也会找到相同的nums[k]。这种情况也应该跳过。
参考代码如下:
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums.sort()
L, res = len(nums), []
for i in range(L-2):
if i > 0 and nums[i] == nums[i-1]:
continue
target = -1 * nums[i]
j,k = i + 1, L - 1
while j<k:
if nums[j]+nums[k] == target:
res.append([nums[i], nums[j], nums[k]])
j = j + 1
while j<k and nums[j] == nums[j-1]:
j = j + 1
elif nums[j] + nums[k] < target:
j = j + 1
else:
k = k - 1
return res
如果您有更好的解法,欢迎讨论。
ps:如果您有好的建议,欢迎交流 :-D,也欢迎访问我的个人博客:tundrazone.com