为了账号安全,请及时绑定邮箱和手机立即绑定

18. 四数之和

标签:
大数据

给定一个包含n 个整数的数组nums和一个目标值target,判断nums中是否存在四个元素 a,b,c和 d,使得a + b + c + d的值与target相等?找出所有满足条件且不重复的四元组。

注意:

答案中不可以包含重复的四元组。

示例:

给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

满足要求的四元组集合为:

[

  [-1,  0, 0, 1],

  [-2, -1, 1, 2],

  [-2,  0, 0, 2]

]

class Solution:

    def fourSum(self, nums, target):

        """

        :type nums: List[int]

        :type target: int

        :rtype: List[List[int]]

        """

        res, dicti = set(), {}

        numLen = len(nums)

        nums.sort()

        for i in range(numLen):

            for j in range(i+1, numLen):

                key = nums[i] + nums[j]

                if key not in dicti.keys():

                    dicti[key] = [(i,j)]

                else:

                    dicti[key].append((i,j))

        for i in range(numLen):

            for j in range(i+1, numLen-2):

                exp = target - nums[i] -nums[j]

                if exp in dicti.keys():

                    for tmpIndex in dicti[exp]:

                        if tmpIndex[0] > j:

                            res.add((nums[i], nums[j], nums[tmpIndex[0]], nums[tmpIndex[1]]))

        return [list(i) for i in res]


webp



作者:不爱去冒险的少年y
链接:https://www.jianshu.com/p/cfd5dbb45a0b


点击查看更多内容
TA 点赞

若觉得本文不错,就分享一下吧!

评论

作者其他优质文章

正在加载中
  • 推荐
  • 评论
  • 收藏
  • 共同学习,写下你的评论
感谢您的支持,我会继续努力的~
扫码打赏,你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫,即可进行扫码打赏哦
今天注册有机会得

100积分直接送

付费专栏免费学

大额优惠券免费领

立即参与 放弃机会
意见反馈 帮助中心 APP下载
官方微信

举报

0/150
提交
取消