leetcode刷题笔记18_四数之和

最新推荐文章于 2024-07-22 16:27:42 发布

椰奶糕

最新推荐文章于 2024-07-22 16:27:42 发布

阅读量129

点赞数

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/lll2135/article/details/126283012

版权

题目

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按任意顺序返回答案。

示例

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

解法

还是排序加双指针的方法，与三数之和类似，固定前两个。

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        n=len(nums)
        if not n and n <4:
            return []
        nums.sort()
        ans=[]
        for i in range(n-3):
            if nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target:
                break
            if nums[n-4]+nums[n-3]+nums[n-2]+nums[n-1]<target:
                break
            if nums[i]==nums[i-1] and i!=0:
                continue
            for j in range(i+1,n-2):
                if nums[j]==nums[j-1] and j>i+1:
                    continue
                L=j+1
                R=n-1
                while L<R:
                    if nums[i]+nums[j]+nums[L]+nums[R]==target:
                        ans.append([nums[i],nums[j],nums[L],nums[R]])
                        while L<R and nums[L]==nums[L+1] :
                            L+=1
                        while L<R and nums[R-1]==nums[R] :
                            R-=1
                        L+=1
                        R-=1
                    elif nums[i]+nums[j]+nums[L]+nums[R]>target:
                        R=R-1
                    else:
                        L=L+1
        return ans