题目的要求如下。
● 找到3个数相加等于0,而不是“两数之和”中的不定值target,因此本题相当于是个特例。
● 题目要求返回的是数本身,而不是索引值。
● 这道题存在多个答案,与上一题不同,本题要求返回所有可能的答案。● 答案中不可以包含重复的三元组,所以需要考虑去重。
思路
由于这道题要求的不再是返回索引值,因此先排序,然后使用双指针的思路是可行的。具体算法是先对原数组进行一次排序,然后一层循环固定一个元素,循环内部利用双指针找出剩下的两个元素,这里要特别注意需要去重。上述算法除去排序部分的时间复杂度为O(n2),相比之下排序过程不会成为性能瓶颈。
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
n=len(nums)
nums.sort()
res=[]
for i in range (n-2):
if i>0 and nums[i]==nums[i-1]:
continue
l=i+1
r=n-1
while(l<r):
if(nums[i]+nums[l]+nums[r]<0):
l+=1
elif(nums[i]+nums[l]+nums[r]>0):
r-=1
else:
res.append([nums[i],nums[l],nums[r]])
while(l<r and nums[l]==nums[l+1]):
l+=1
while(l<r and nums[r]==nums[r-1]):
r-=1
l+=1
r-=1
return res
复杂度分析
● 时间复杂度:上述代码for循环内部虽然有两个while循环,但是这两个while循环也仅仅会扫描一遍数组(最里面的while循环只是跳过一些重复的元素而已),因此总的时间复杂度仍然是O(n2),而不是O(n3),其中n为数组长度。
● 空间复杂度:不确定,取决于内部排序算法的具体实现。