菜鸟的LeetCode之旅-015三数之和

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
     满足要求的三元组集合为：
     [
          [-1, 0, 1],
          [-1, -1, 2]
     ]

方法一
开始的想法是暴力破解，利用先将数组排序，然后利用三重for循环来找到符合要求的a，b，c。这种方法时间复杂度为O(n^3)，在leetcode中提交果然时间超时为此改进方法一，得到方法二。

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()              #数组排序
        l = len(nums)
        num = []                #存储和为0的列表
        if l < 3:               #排除长度小于3的数组
            return num
        for i in range(l-2):
            for j in range(i+1,l-1):
                for k in range(j+1,l):
                    if nums[i] + nums[j] + nums[k] == 0 and [nums[i],nums[j],nums[k]] not in num:
                        num.append([nums[i],nums[j],nums[k]])
        return num

ts = Solution()
nums = [-1, 0, 1, 2, -1, -4]
print(ts.threeSum(nums))

方法二
尝试去掉一重循环，引入another = 0 - a - b，可是结果还是不理想，提交结果依然超时。

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()              #数组排序
        l = len(nums)
        num = []                #存储和为0的列表
        if l < 3:               #排除长度小于3的数组
            return num
        for i in range(l-2):
            for j in range(i+1,l-1):
                another = 0 - nums[i] - nums[j]
                if another in nums[j+1:l] and [nums[i],nums[j],another] not in num:
                    num.append([nums[i],nums[j],another])
        return num

ts = Solution()
nums = [-1, 0, 1, 2, -1, -4]
print(ts.threeSum(nums))