题目描述:
给定一个整数数组 nums 和一个整数目标值 target,请给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
来源:力扣(LeetCode)
算法思路:
方法一: 暴力解法,三重循环,时间复杂度O(n^3) 这里没有去实现
方法二: 排序+双指针 时间复杂度为O(n^2)
1、先对list 数组进行排序,然后开始两层循环
2、第一层循环 i 从0 到 n-3,每次取i后 后面的数据开始循环,采用双指针left=i+1和right=n-1 ,直到left>right 循环结束
3、第一层忽略相邻重复元素的方法 是判断nums[i] 与 nums[i-1]是否相等,相等则跳过。第二层循环忽略相邻重复元素的方法是判断nums[left]==nums[left+1] 和 nums[right]==nums[right-1] 是否相等,相等则跳过。
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums=sorted(nums)
list=[]
length=len(nums)
if length<3:
return list
for i in range(length-2):
if nums[i]>0:
return list
if i>0 and nums[i]==nums[i-1]:
continue
left,right=i+1,length-1
while left<right:
if nums[i]+nums[left]+nums[right]==0:
list.append ([nums[i],nums[left],nums[right]])
while left<right and nums[left]==nums[left+1]:
left=left+1
while left<right and nums[right]==nums[right-1]:
right=right-1
left=left+1
right=right-1
elif nums[i]+nums[left]+nums[right]<0:
left=left+1
else:
right=right-1
return list