算法:三个数之和
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
一开始看到这个题目,我并没有想太多,没有考虑复杂度神马的,就直接写了
class Solution(object):
def threeSum(self,nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
res = []
nums.sort()
N = len(nums)
for i in range(0,N-3):
for j in range(i+1,N-2):
for k in range(j+1,N-1):
if nums[i]+nums[j]+nums[k] == 0:
res.append([nums[i],nums[j],nums[k]])
return res
写完了,一运行发现了,有一种重复的情况根本就去除不了,
算了,加一个筛选吧
res = []
# nums.sort()
N = len(nums)
for i in range(0,N-3):
for j in range(i+1,N-2):
for k in range(j+1,N-1):
if nums[i]+nums[j]+nums[k] == 0:
res.append([nums[i],nums[j],nums[k]])
for x in range(0,len(res)-1):
for y in range(0,len(res)-1):
if res[x][1]==res[y][0] and res[x][2] == res[y][1]:
del res[y]
return res
好丑啊,还是想一想算了好好写吧
nums.sort()
res =[]
i = 0
for i in range(len(nums)):
if i == 0 or nums[i]>nums[i-1]:
l = i+1
r = len(nums)-1
while l < r:
s = nums[i] + nums[l] +nums[r]
if s ==0:
res.append([nums[i],nums[l],nums[r]])
l +=1
r -=1
while l < r and nums[l] == nums[l-1]:
l += 1
while r > l and nums[r] == nums[r+1]:
r -= 1
elif s>0:
r -=1
else :
l +=1
return res
浙公网安备 33010602011771号