leetcode.array--15. 3Sum

题目：15. 3Sum
问题描述：https://leetcode.com/problems/3sum/description/
大致意思是：给出一个列表，找出其中三个数，使得三个数相加结果等于0，要求找到所有组合。

刚刚写过一个在列表中找出两个数相加等于定值的题目，自然又想到了用相同的方法解题。两个数字相加等于定值的题目中，可以用哈希表解决，时间复杂度O(n)，如果这个题目也采用相同的思路，时间复杂度将是O(n^2)。比较幸运的是这种思路没有超时，不过代码跑了一秒多的时间。

Python：

class Solution(object):    def threeSum(self, nums):        """        :type nums: List[int]        :rtype: List[List[int]]        """        res = []        if len(nums) <= 2:            return res        nums.sort()        lens = len(nums)        if nums[0]==0 and nums[lens-1]==0:            return [[0,0,0]]        dicres = {}        for i in range(lens):            if nums[i]>0:                break            dic = {}            elnum = 0 - nums[i]            for j in range(i + 1, lens):                if nums[j] in dic:                    temp = [nums[i], nums[dic[nums[j]]], nums[j]]                    #temp.sort()                    tempstr=str(temp)                    if tempstr not in dicres:                        res.append(temp)                        dicres[tempstr] = 1                else:                    dic[elnum - nums[j]] = j        return res

后来看了别人的代码，不用dict也可以：先排序，定下一个数字(nums[i])后从两头(left=i+1,right=len-1)开始计算，如果三个数的和大于0，right向左移动(right-=1)，若小于0，则left向右移动(left+=1)，等于0的时候就是我们想要的结果，这样时间复杂度为O(n^2)，实际运行时间不到一秒，要比上面快不少。

Python:

class Solution(object):    def threeSum(self, nums):        """        :type nums: List[int]        :rtype: List[List[int]]        """        res=[]        lens=len(nums)        if lens<=2:            return res        nums.sort()        if nums[0]>0:            return res        for i in range(lens):            if nums[i]>0:                break            if i>0 and nums[i]==nums[i-1]:                continue            left=i+1            right=lens-1            while left<right:                if nums[left]+nums[right]+nums[i]>0:                    right-=1                elif nums[left]+nums[right]+nums[i]<0:                    left+=1                else:                    temp=[nums[i],nums[left],nums[right]]                    res.append(temp)                    right -= 1;left+=1                    while left<right and nums[left]==nums[left-1]:                        left+=1                    while left<right and nums[right]==nums[right+1]:                        right-=1        return res