【leetcode】15. 三数之和

三重循环+去重

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        length = len(nums)
        l = []
        p1, p2, p3 = 0, 0, 0
        while p1 < length:
            p2 = p1 + 1
            while p2 < length:
                p3 = p2 + 1
                while p3 < length:
                    if nums[p1] + nums[p2] + nums[p3] == 0:
                        l.append([nums[p1], nums[p2], nums[p3]])
                    p3 += 1
                p2 += 1
            p1 += 1
        n = []
        nums_set = []
        l2 = []
        for i, v in enumerate(l):
            if set(v) not in nums_set:
                nums_set.append(set(v))
            else:
                n.append(i)
        for i in range(len(l)):
            if i not in n:
                l2.append(l[i])
        return l2

排序+双指针

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        length = len(nums)
        ans = list()
        if length < 3:
            return []
        for i in range(length):
            if nums[i] > 0:
                break
            if i > 0 and nums[i] == nums[i - 1]:
                continue
            L = i + 1
            R = length - 1
            while L < R:
                if nums[i] + nums[L] + nums[R] == 0:
                    ans.append([nums[i], nums[L], nums[R]])
                    while L < R and nums[L] == nums[L + 1]:
                        L += 1
                    while L < R and nums[R] == nums[R - 1]:
                        R -= 1
                    L += 1
                    R -= 1
                elif nums[i] + nums[L] + nums[R] < 0:
                    L += 1
                else:
                    R -= 1
        return ans

总结

1、方法一通过三重循环获取数组，但是题目要求最终结果不能存在重复，因此需要先记录列表重复元素位置，再将不重复数组，复制至新列表中。但是由于时间复杂度为O(n³)，结果超时了。
2、方法二：该题重点是如何去除列表中的重复数组，为此的解决方法：首先先对列表元素排序，然后再进行遍历，在第i次循环中，若该元素与（i-1）的元素相同，则跳过第i个元素，以此去除重复的数组。
  双指针：我们在第二层循环内，寻找第2，3个元素的过程中，两个指针分别指向两侧，通过nums[i] + nums[L] + nums[R] == 0，判断是否要将三个元素放入列表中，以及是否移动指针。第二层循环一下子找到了两个元素，所以只需两层循环，时间复杂度为O(n²)。