Python | Leetcode Python题解之第18题四数之和

Mopes__

于 2024-04-09 05:29:25 发布

阅读量434

点赞数 15

分类专栏：分享文章标签： Python Leetcode 题解

本文链接：https://blog.csdn.net/Mopes__/article/details/137532781

版权

分享专栏收录该内容

244 篇文章 2 订阅

订阅专栏

本文介绍了如何使用Python实现一个名为`fourSum`的函数，用于解决给定整数数组中的四数之和问题，通过排序和双指针法寻找满足特定目标值的四元组解。

摘要由CSDN通过智能技术生成

题目：

题解：

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        quadruplets = list()
        if not nums or len(nums) < 4:
            return quadruplets
        
        nums.sort()
        length = len(nums)
        for i in range(length - 3):
            if i > 0 and nums[i] == nums[i - 1]:
                continue
            if nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target:
                break
            if nums[i] + nums[length - 3] + nums[length - 2] + nums[length - 1] < target:
                continue
            for j in range(i + 1, length - 2):
                if j > i + 1 and nums[j] == nums[j - 1]:
                    continue
                if nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target:
                    break
                if nums[i] + nums[j] + nums[length - 2] + nums[length - 1] < target:
                    continue
                left, right = j + 1, length - 1
                while left < right:
                    total = nums[i] + nums[j] + nums[left] + nums[right]
                    if total == target:
                        quadruplets.append([nums[i], nums[j], nums[left], nums[right]])
                        while left < right and nums[left] == nums[left + 1]:
                            left += 1
                        left += 1
                        while left < right and nums[right] == nums[right - 1]:
                            right -= 1
                        right -= 1
                    elif total < target:
                        left += 1
                    else:
                        right -= 1
        
        return quadruplets