[LeetCode] 3Sum

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[LeetCode] 3Sum相关的知识,希望对你有一定的参考价值。

题目内容

Given an array S of n integers, are there elements a, b, c in S 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.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

题目思路

本题难度:medium

本题是在2sum的基础上延伸的。在2sum中,target是一个函数的输入值。而在这里处理3sum的时候,使用一个循环遍历nums数组,分别让target= -(nums[i])。然后就可以将问题转换为2sum,只不过比2sum多了一个for循环。

这个方法可以推广到一般的ksum的情况,ksum需要k-2次循环。

这里使用的方法是在2sum中提到的双指针法。需要说明的是,在本题中,提到了结果不允许重复,因此在移动指针的时候,需要判断是否重复。

Python代码

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums=sorted(nums)
        _len=len(nums)
        res=[]
        for i in xrange(_len-2):
            if nums[i]==nums[i-1] and i>0:
                continue
            target=nums[i]
            left=i+1
            right=_len-1
            while left<right:
                _sum=nums[left]+nums[right]
                if _sum+target==0:
                    res.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
                elif _sum+target<0:
                    left+=1
                elif _sum+target>0:
                    right-=1
        return res

 

以上是关于[LeetCode] 3Sum的主要内容,如果未能解决你的问题,请参考以下文章

leetcode 15. 3Sum

leetcode16 3Sum Closest

[Leetcode] 3Sum

4Sum -- LeetCode

LeetCode(16):3Sum Closest

Leetcode 15. 3Sum