ITPub博客

首页 > IT职业 > IT生活 > 18. 四数之和

18. 四数之和

IT生活 作者:snoopyzjs 时间:2021-09-09 10:13:04 0 删除 编辑

给定一个包含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]


图片描述



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


来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/1414/viewspace-2818350/ ,如需转载,请注明出处,否则将追究法律责任。

下一篇: scala隐式转换
请登录后发表评论 登录
全部评论

注册时间:2019-04-15

  • 博文量
    104
  • 访问量
    1015