LeetCode18. 四数之和 python3

Victordas

于 2018-09-26 12:25:47 发布

阅读量777

点赞数

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.csdn.net/Victordas/article/details/82851635

Python 专栏收录该内容

47 篇文章

订阅专栏

本文深入探讨了在给定数组和目标值的情况下，寻找所有不重复的四元组，使其和等于目标值的算法实现。通过排序和双指针技巧，避免了重复解并提高了搜索效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一个包含 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]]
        """
        nums.sort()
        num_save = []

        if len(nums) < 4 or nums == []:
            return []
        else:
            if len(nums) == 4 and sum(nums) == target:
                num_save.append(nums)
                return num_save
            else:
                for i in range(0, len(nums)-2):
                    for j in range(i+1, len(nums)-1):
                        l = j+1
                        r = len(nums) - 1
                        while l < r:
                            sums = nums[i] + nums[j] + nums[l] + nums[r]
                            if sums == target:
                                num_save.append([nums[i], nums[j], nums[l], nums[r]])
                                break
                            elif sums < target:
                                l += 1
                            else:
                                r -= 1

                num_saves = list(set([tuple(t) for t in num_save]))
                return num_saves