力扣 ——3Sum python (三数之和)实现

Posted hehe哒

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了力扣 ——3Sum python (三数之和)实现相关的知识,希望对你有一定的参考价值。

题目描述:

中文:

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

英文:

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

The solution set must not contain duplicate triplets.

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        result = []
        if len(nums) < 3:
            return result
        nums.sort()  
        for i in range(0, len(nums)):
            j = i + 1
            k = len(nums) - 1
            if i > 0 and nums[i] == nums[i - 1]:
                continue  #  
            while j < k:
                sum = nums[i] + nums[j] + nums[k]
                if sum == 0:
                    result.append((nums[i], nums[j], nums[k]))
                    k -= 1
                    j += 1
                    while(nums[j] == nums[j - 1] and nums[k] == nums[k + 1] and j < k): 
                        j += 1
                elif sum > 0:
                    k -= 1
                    while(nums[k] == nums[k + 1] and j < k):
                        k -= 1
                else:
                    j += 1
                    while(nums[j] == nums[j - 1] and j < k):
                        j += 1
        return list(set(result))

 

 

题目来源:力扣

以上是关于力扣 ——3Sum python (三数之和)实现的主要内容,如果未能解决你的问题,请参考以下文章

力扣16-最接近的三数之和&力扣18-四数之和

[Lintcode 3sum]三数之和(python,二分)

[LintCode] 3Sum 三数之和

leetcode 15. 三数之和(sort+双指针)

LeetCode 15. 3Sum(三数之和)

[LeetCode] 15. 3Sum 三数之和