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