给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
NOTE:在计算过程中需要进行排重。
class Solution(object): def threeSum(self, nums): """ :type nums: List[int] :rtype: List[List[int]] """ target = 0 ret = [] nums.sort() for p1 in range(len(nums) - 2): if p1 > 0 and nums[p1] == nums[p1 - 1]: continue left = p1 + 1 right = len(nums) - 1 while left < right: a = nums[p1] + nums[left] + nums[right] - target if a == 0: ret.append([nums[p1], nums[left], nums[right]]) while left < right and nums[left + 1] == nums[left]: left = left + 1 left = left + 1 while right > left and nums[right - 1] == nums[right]: right = right - 1 right = right - 1 elif a < 0: left = left + 1 else: right = right - 1 return ret