题目描述:
18. 四数之和
给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。
注意:答案中不可以包含重复的四元组。
示例 1:
输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2:
输入:nums = [], target = 0
输出:[]
代码:
class Solution:
def fourSum(self, nums: list, target: int) -> list:
n = len(nums)
nums.sort()
res = []
flag = 0
for k in range(n-3):
if k > 0 and nums[k] == nums[k-1]:
continue
for l in range(k+1, n-2):
if l > 1 and nums[l] == nums[l-1] and l-1 != k:
continue
i = l+1
j = n-1
while i < j:
sum = nums[i]+nums[j]+nums[l]+nums[k]
if sum < target:
i += 1
elif sum > target:
j -= 1
else:
res.append([nums[k], nums[l], nums[i], nums[j]])
i += 1
j -= 1
while i < j and nums[i] == nums[i-1]:
i += 1
while i < j and nums[j] == nums[j+1]:
j -= 1
return res
if __name__ == '__main__':
s = Solution()
print(s.fourSum([-4,-3,-2,-1,0,0,1,2,3,4],0))
解题思路:
排序+双指针
基本思路:这题解法和三数之和一样,比三数之和多了一重循环,第一重循环k从0到n-4,第二重循环 l 从k+1到n-3(下标从0开始,所以是到n-3)。接下来就是和三数之和一样的解法了,双指针,i 从 l+1 开始,j从n-1开始,计算和sum,小于target则i+1,大于target则j-1,等于target把该组数加入到结果中,执行i+1,j-1,选取下一组数。
去重:四数之和的去重第一重循环和里面的双指针去重方式是一样的,直接与上一个状态相比较,相同就跳过。第二重循环去重需要判断一下 l 是否大于k+1再进行比较,这样才能确保进行比较的是选取的四元组中的第二个数,而不是与k对应的值(即该轮选取的四元组的第一个数)比较。
时间复杂度:双重循环加双指针,时间复杂度是O(N^3),排序的时间复杂度是O(nlogn),所以总的时间复杂度是O(N ^3+nlogn),小的数量级忽略,即O(N ^3)。