标签：nums 三数 2020.6 resultSet len Note middle endInd first

题目

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

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

 

示例：

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

1. 3重循环暴力法
2. 官方解法，在3种循环暴力法的基础上加入双指针，将复杂度从O(N^{3）变为O(N}2)。
   首先注意两点，第一和为0，第二无重复。那么首先将序列排序后，开始三重循环枚举a,b,c，此时a<=b<=c是必然成立的；在确定a的情况下，若有a+b+c=0， 那b此时想右移一位得到b1后，令a+b1+c1=0的c1必然比c更小，所以c的取值c1应该是向c的左边序列找。因此将b和c的这两重循环是高度依赖的，可以通过双指针（一个自左往右，一个自右往左）将搜索复杂度从O(N^2)变为O(N)。

代码

• 思路1：

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        resultSet = {}
        for i in range(len(nums)):
            first = nums[i]
            for j in range(i+1, len(nums)):
                middle = nums[j]
                for k in range(j+1, len(nums)):
                    end = nums[k]
                    if (first + middle + end) == 0:
                        tmp = [first, middle, end]
                        tmp.sort()
                        first1,middle1,end1 = tmp
                        if first1 in resultSet:
                            if middle1 in resultSet[first1]:
                               resultSet[first1][middle1].add(end1)
                            else:
                                resultSet[first1][middle1] = set([end1])
                        else:
                            resultSet[first1]={middle1: set([end1])}
        resultList = []
        for key,v in resultSet.items():
            for k2,v2 in v.items():
                while len(v2)>0:
                    resultList.append([key, k2, v2.pop()])
        return resultList

毫无疑问，在这种平台上，这种无效率的做法是会超时的。

2. 思路2：

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        resultSet = []
        nums.sort()
        for i in range(len(nums)):
            if i!=0 and (nums[i] ==  nums[i-1]):
                continue

            first = nums[i]
            endInd = len(nums) - 1

            for j in range(i+1, len(nums)):
                if j!=i+1 and (nums[j] == nums[j-1]):
                    continue

                middle = nums[j]

                while endInd>j and  (first + middle + nums[endInd]) > 0:
                    endInd -= 1
                if endInd>j and (first + middle + nums[endInd]) == 0:
                    resultSet.append([first, middle, nums[endInd]])
                
        return resultSet

标签：nums,三数,2020.6,resultSet,len,Note,middle,endInd,first
来源： https://www.cnblogs.com/immortalBlog/p/13098544.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。