解题思路-leetcode第十八题:四数之和
题目描述:给定一个包含 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]
]
本题采用和三数之和之和同样的解法,只是在最外层又加了一层循环。首先排除特殊情况,然后对列表升序排序,然后是第一层循环,在循环内,i值不变,此时,若i和i后的三个数之和大于target,则i之后的数不会再出现解,跳出循环,若i与最后三个数之和小于target,则此i无解,跳入下一循环,若当前i代表的值与下一个值相同,则跳过此i,进入下一循环,若上述条件都不满足,则进入内嵌的for循环,在此循环内,遍历i之后的数,与上一层循环一样,i,j的值不变,然后判断是此j以后都无解还是仅此j无解,然后再排除重复解,之后 则与三数之和思路相同,设置两个指针,遍历j之后的数,当遍历到的四个数满足条件时,添加进存储结果的res列表,最后返回该列表。代码如下:
class Solution:
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
n=len(nums)
if(not nums or n<4):
return []
nums.sort()
res=[]
for i in range(n-3):
if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target):
break
if(nums[i]+nums[-1]+nums[-2]+nums[-3]<target):
continue
if(i>0 and nums[i]==nums[i-1]):
continue
for j in range(i+1,n-2):
if(nums[i]+nums[j]+nums[j+1]+nums[j+2]>target):
break
if(nums[i]+nums[j]+nums[-1]+nums[-2]<target):
continue
if(j-i>1 and nums[j]==nums[j-1]):
continue
L=j+1
R=n-1
while(L<R):
if(nums[i]+nums[j]+nums[L]+nums[R]==target):
res.append([nums[i],nums[j],nums[L],nums[R]])
while(L<R and nums[L]==nums[L+1]):
L=L+1
while(L<R and nums[R]==nums[R-1]):
R=R-1
L=L+1
R=R-1
elif(nums[i]+nums[j]+nums[L]+nums[R]>target):
R=R-1
else:
L=L+1
return res
提交后,通过。