【leetcode】15. 三数之和

【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²)。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值