[leetcode]三数之和

最新推荐文章于 2025-09-05 11:17:42 发布

weixin_33898876

最新推荐文章于 2025-09-05 11:17:42 发布

阅读量66

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法 python

原文链接：https://my.oschina.net/bugyang/blog/1931670

本文提供了一个Python实现的方法来解决寻找数组中是否存在三个数相加等于零的问题，并给出了具体的解决方案及代码示例。

2019独角兽企业重金招聘Python工程师标准>>>

题目：

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

代码：

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        res_list = []
        nums_len = len(nums)
        nums = sorted(nums)
        i = 0
        while i < nums_len - 2:
            j = i + 1
            k = nums_len - 1
            while j < k:
                # print "%s %s %s" % (nums[i], nums[j], nums[k])
                if nums[i] + nums[j] + nums[k] == 0:
                    new_list = [nums[i], nums[j], nums[k]]
                    res_list.append(new_list)
                    j = j + 1
                    k = k - 1
                    while nums[j] == nums[j - 1] and j < k:
                        j = j + 1
                    while nums[k] == nums[k + 1] and j < k:
                        k = k - 1
                elif nums[i] + nums[j] + nums[k] < 0:
                    j = j + 1
                elif nums[i] + nums[j] + nums[k] > 0:
                    k = k - 1
            i = i + 1
            while nums[i] == nums[i - 1] and i < nums_len - 2:
                i = i + 1
        return res_list

转载于:https://my.oschina.net/bugyang/blog/1931670