·题目
给你一个整数数组 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 。
·思路
超时的思路:
1.两个循环暴力求解,再找另外一个数字————>超时O(n^3)
2.将原有数组分为正负零他,再找,还是超时O(n^3)
—————记录一下gpt给的思路————
技巧在于:先排序再用双指针滑动
1.排序的好处在于,可以有规律的滑动
2.双指针可有有效减少时间复杂度
首先对数组进行排序(O(n log n)),然后遍历数组(O(n)),并在每次遍历中使用双指针进行线性搜索(O(n))。由于排序的时间复杂度是主导项,因此整体时间复杂度可以认为是O(n log n),但在实际情况下,双指针的遍历是算法的主要瓶颈,因此通常认为是O(n^2)。
3.使用continue去除重复元素
·代码
class Solution(object):def threeSum(self, nums):nums.sort()n = len(nums)result = []for i in range(n-2):#指定一个元素,双指针针对该元素进行遍历left = i + 1right = n - 1if i > 0 and nums[i] == nums[i - 1]:continue #当出现相同数字的时候,跳出循环,并跳到下一次迭代while left < right:cur_add = nums[i] + nums[left] + nums[right] #记录和if cur_add == 0:result.append([nums[i] , nums[left] , nums[right]])while left < right and nums[left ] == nums[left + 1]:left += 1 #对于同一个指定元素,当其中两个数都相同,那么为0的化,另外一个也相同,所以直接跳过while left < right and nums[right- 1] == nums[right]: #同理right -= 1left += 1right -= 1 elif cur_add < 0:left += 1else:right -= 1return result