·题目
给你一个整数数组 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 + 1
right = n - 1
if 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 -= 1
left += 1
right -= 1
elif cur_add < 0:
left += 1
else:
right -= 1
return result