三重循环+去重
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
length = len(nums)
l = []
p1, p2, p3 = 0, 0, 0
while p1 < length:
p2 = p1 + 1
while p2 < length:
p3 = p2 + 1
while p3 < length:
if nums[p1] + nums[p2] + nums[p3] == 0:
l.append([nums[p1], nums[p2], nums[p3]])
p3 += 1
p2 += 1
p1 += 1
n = []
nums_set = []
l2 = []
for i, v in enumerate(l):
if set(v) not in nums_set:
nums_set.append(set(v))
else:
n.append(i)
for i in range(len(l)):
if i not in n:
l2.append(l[i])
return l2
排序+双指针
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums.sort()
length = len(nums)
ans = list()
if length < 3:
return []
for i in range(length):
if nums[i] > 0:
break
if i > 0 and nums[i] == nums[i - 1]:
continue
L = i + 1
R = length - 1
while L < R:
if nums[i] + nums[L] + nums[R] == 0:
ans.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
elif nums[i] + nums[L] + nums[R] < 0:
L += 1
else:
R -= 1
return ans
总结
1、方法一通过三重循环获取数组,但是题目要求最终结果不能存在重复,因此需要先记录列表重复元素位置,再将不重复数组,复制至新列表中。但是由于时间复杂度为O(n³),结果超时了。
2、方法二:该题重点是如何去除列表中的重复数组,为此的解决方法:首先先对列表元素排序,然后再进行遍历,在第i次循环中,若该元素与(i-1)的元素相同,则跳过第i个元素,以此去除重复的数组。
双指针:我们在第二层循环内,寻找第2,3个元素的过程中,两个指针分别指向两侧,通过nums[i] + nums[L] + nums[R] == 0,判断是否要将三个元素放入列表中,以及是否移动指针。第二层循环一下子找到了两个元素,所以只需两层循环,时间复杂度为O(n²)。