给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请
你返回所有和为 0
且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]] 解释: nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。 nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。 nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。 不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1] 输出:[] 解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0] 输出:[[0,0,0]] 解释:唯一可能的三元组和为 0 。
python代码答案:
class Solution(object):
def threeSum(self, nums):
result=[]
for i in range(0,len(nums)):
for j in range(i+1,len(nums)):
a=nums[i]+nums[j]
complement=-a
for l in range(j+1,len(nums)):
if nums[l]==complement:
if sorted([nums[i], nums[j], nums[l]]) not in result:
result.append([nums[i], nums[j], nums[l]])
break
return result
if __name__=='__main__':
nums=[-1, 0, 1, 2, -1, -4]
solution=Solution()
result1=solution.threeSum(nums)
result = sorted(set(tuple(sorted(x)) for x in result1))
print(result)
简单思路:
对于这道题,直接用嵌套的三层循环,实现将3个数相加,分解成先实现让两个数相加a=nums[i]+nums[j],因为有三个数相加为这个约束条件因此,可以直接在第三层循环范围是前两个数没有循环到剩下的那些。
循环的时候切记数组的边界,不要访问超出数组元素的边界。
然后,找第三个数nums[l],让a3==-sum,判断在范围内是否存在该元素。我用了
complement=-a
for l in range(j+1,len(nums)):
if nums[l]==complement:
代码来判断,
如果直接在第三次循环中用:
if num[l]==-a:
相关操作……
一旦if num[l]==-a:成立这样会不能跳出循环
重点:在去重的时候,正确使用了set
和tuple
函数,实现正确地去除排序无关的重复元素。
问题的关键在于,当对元组进行排序并转换为集合时,在每个子列表中单独进行去重,而不是在整个结果列表中进行去重。因此,即使两个子列表包含相同的元素,但只要它们的相对顺序不同,它们就不会被视为重复。
为了解决这个问题,你可以在将元组添加到结果列表之前,先对元组进行排序,然后再将其转换为集合。这样,排序无关的重复元素将会被去除。
if sorted([nums[i], nums[j], nums[l]]) not in result:
result.append([nums[i], nums[j], nums[l]])
但是悲哀的是,这种简单粗暴的方法,时间复杂度是O(n^3),超时了,~(QAQ)~
泪奔了QAQ
OK,那再改进改进,我直接用双指针将时间复杂度降为:O(n^2)
和上面代码思路一样,数组和指针联系真的很多,一般用循环解决数组问题,指针也是可以的啦。
各位,大老爷代码改进如下:
class Solution(object):
def threeSum(self, nums):
nums.sort() # 先对数组进行排序
result = []
for i in range(len(nums)-2): # 第一个指针
if i > 0 and nums[i] == nums[i-1]: # 去重,跳过重复的元素
continue
left, right = i+1, len(nums)-1 # 初始化双指针
while left < right: # 双指针方法
total = nums[i] + nums[left] + nums[right]
if total == 0: # 如果和为0,则找到了一个解
result.append([nums[i], nums[left], nums[right]])
while left < right and nums[left] == nums[left+1]: # 去重,跳过重复的元素
left += 1
while left < right and nums[right] == nums[right-1]: # 去重,跳过重复的元素
right -= 1
left += 1
right -= 1
elif total < 0: # 如果和小于0,左指针右移
left += 1
else: # 如果和大于0,右指针左移
right -= 1
return result
if __name__=='__main__':
nums=[-1, 0, 1, 2, -1, -4]
solution=Solution()
result1=solution.threeSum(nums)
result = sorted(set(tuple(sorted(x)) for x in result1)) # 去重并排序结果
print(result)
~(QAQ)~今天的分析到这里啦,小菜鸟继续上线,加油打气!!!!啦啦啦啦啦啦啦啦,哈哈哈哈哈哈哈哈哈!如果你喜欢关于对编程解题感兴趣的话,请关注我哦~~~
我将持续发力~~~~~~~