LeetCode 热题 HOT 100之三数之和
题目15:给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
方案:排序加双指针,先对数组进行排序,对每个位置i,找到符合条件的另两个位置,为了保证结果中不发生重复,需要加一些判断条件。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
ans = []
nums.sort()
for i in range(len(nums)):
#防止重复
if i > 0 and nums[i] == nums[i - 1]:
continue
target = -nums[i]
k = len(nums) - 1
for j in range(i + 1, len(nums)):
#防止重复
if j > i + 1 and nums[j] == nums[j - 1]:
continue
while j < k and nums[j] + nums[k] > target:
k -= 1
if j == k:
break
if nums[k] + nums[j] == target:
ans.append([nums[i], nums[j], nums[k]])
return ans
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。