题目15—三数之和

原创已于 2024-03-25 20:13:32 修改 · 204 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

于 2024-03-19 20:49:06 首次发布

该篇文章介绍了如何解决一个编程问题，即在给定的整数数组中找到所有和为0且不重复的三元组，通过先对数组排序并采用双指针法进行搜索。示例数组nums=[-1,0,1,2,-1,-4]的结果是[[-1,-1,2],[-1,0,1]]。

题目来源于LeetCode

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

# 三数之和 a+b+c=0

nums = [-1, 0, 1, 2, -1, -4]

nums = sorted(nums)

arr = list()
n = len(nums)
for i in range(n):
    if i > 0 and nums[i] == nums[i - 1]:
        continue
    target = -nums[i]
    end = n - 1
    for start in range(i + 1, n):
        if start > i + 1 and nums[start] == nums[start - 1]:
            continue
        while start < end and nums[start] + nums[end] > target:
            end -= 1
        if start == end:
            break
        if nums[start] + nums[end] == target:
            arr.append([nums[i], nums[start], nums[end]])

print(arr)  # [[-1, -1, 2], [-1, 0, 1]]