网易笔试题-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_51570574/article/details/133868925

文章介绍了三种方法解决给定整数数组中是否存在三个元素之和为零的问题：暴力解法通过遍历所有可能组合并去重；双指针法利用两个指针分别从数组两端向中间移动；哈希表法通过存储中间元素的和及其索引，避免重复计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。
注意：答案中不可以包含重复的三元组。
示例 1：
输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

示例 2：
输入：nums = []
输出：[]

示例 3：
输入：nums = [0]
输出：[]

方法一：暴力解法

找出所有符合条件的三元组，排序后去除相同三元组；

注意：例如[-1,0,1],[-1,1,0]属于重复三元组

def three_sum_bruteforce(nums):
    result = []

    for i in range(len(nums) - 2):
        for j in range(i + 1, len(nums) - 1):
            for k in range(j + 1, len(nums)):
                if nums[i] + nums[j] + nums[k] == 0:
                    current_result = [nums[i], nums[j], nums[k]]
                    current_result.sort()  # 为了避免重复，排序后加入结果集
                    if current_result not in result:
                        result.append(current_result)

    return result

# 示例
nums = [-1, 0, 1, 2, -1, -4]
result = three_sum_bruteforce(nums)
print(result)

运行结果：

方法二：

双指针

先将数组进行排序；

左指针指向i+1的位置，右指针指向数组末尾；

循环

for i in range(len(nums) - 2):

总和小于0，左指针右移1

大于0，右指针左移

等于0，记录，加入结果集。并且跳过相同元素

while left < right and nums[left] == nums[left + 1]:
    left += 1
while left < right and nums[right] == nums[right - 1]:
    right -= 1
left += 1
right -= 1

返回结果输出；

代码如下：

def three_sum(nums):
    result = []
    nums.sort()

    for i in range(len(nums) - 2):
        if i > 0 and nums[i] == nums[i - 1]:
            continue

        left, right = i + 1, len(nums) - 1

        while left < right:
            total = nums[i] + nums[left] + nums[right]

            if total < 0:
                left += 1
            elif total > 0:
                right -= 1
            else:
                result.append([nums[i], nums[left], nums[right]])
                while left < right and nums[left] == nums[left + 1]:
                    left += 1
                while left < right and nums[right] == nums[right - 1]:
                    right -= 1
                left += 1
                right -= 1

    return result

# 示例
nums = [-1, 0, 1, 2, -1, -4,2,4,-3]
result = three_sum(nums)
print(result)

方法三：哈希表

def three_sum(nums):
    result = []
    nums.sort()

    for i in range(len(nums) - 2):
        if i > 0 and nums[i] == nums[i - 1]:
            continue

        target = -nums[i]
        left, right = i + 1, len(nums) - 1
        seen = set()

        while left < right:
            current_sum = nums[left] + nums[right]

            if current_sum == target and (nums[left], nums[right]) not in seen:
                result.append([nums[i], nums[left], nums[right]])
                seen.add((nums[left], nums[right]))
                left += 1
                right -= 1
            elif current_sum < target:
                left += 1
            else:
                right -= 1

    return result

# 示例
nums = [-1, 0, 1, 2, -1, -4]
result = three_sum(nums)
print(result)