持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第15天,点击查看活动详情
给你一个由 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
你可以按 任意顺序 返回答案 。
示例 1:
输入: nums = [1,0,-1,0,-2,2], target = 0
输出: [[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2:
输入: nums = [2,2,2,2,2], target = 8
输出: [[2,2,2,2]]
提示:
1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109
解题思路
本题和 leetcode-15-三数之和 类似,只不过变成了四数之和且目标和值不再是固定的 0
,而是传入的 target
。
首先对数组进行排序,这样做的好处是可以极大缩小查找的范围,比如如果当前位置的值大于 target
且 当前位置的值 >=0
,那么它和它后面的值肯定无法组合成目标数组。
还有一个好处是如果当前元素和前一个元素相同,则跳过当前元素的处理,因为结果数组中不允许有同样的组合,如果当前元素和它前面的元素相同,则如果前面元素有符合条件的组合,它也会有,而且会相同,所以跳过。
利用以上两种策略,就可以极大程度的减少操作次数。
代码实现
var fourSum = function(nums, target) {
const len = nums.length
if(len<4){
return []
}
let res = []
nums.sort((a,b) => a-b)
for(let i = 0;i<len-3;i++){
if(nums[i]>target && nums[i]>=0){
return res
}
if(i>0 && nums[i]===nums[i-1]){
continue
}
for(let j = i+1;j<len-2;j++){
if(nums[i]+nums[j] > target && nums[j]>=0){
break
}
if(j>i+1 && nums[j]===nums[j-1]){
continue
}
let head = j+1
let end = len-1
while(head<end){
if(end<len-1 && nums[end]===nums[end+1]){
end--
continue
}
if(head>j+1 && nums[head]===nums[head-1]){
head++
continue
}
const sum = nums[i]+nums[j]+nums[head]+nums[end]
if(sum===target){
res.push([nums[i],nums[j],nums[head],nums[end]])
}
if(sum-target>0){
end--
}else{
head++
}
}
}
}
return res
}
至此我们就完成了 leetcode-18-四数之和
如有任何问题或建议,欢迎留言讨论!