给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。
注意:答案中不可以包含重复的四元组
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum
相似题目
思路
这题只是三数之和稍加改一下
三数之和是由i的位置来确定a,然后left从i+1开
始来确定b,right从nums.length-1开始来确定c;
四数之和则多加一重循环,同样i的位置来确定a,但
i+1的位置来确定b,left从j+1的位置来确定c,right从nums.length-1开始来确定d
剩下的就都差不多,包括去重的方法,跟三数之和的方法是一样的直接看代码
代码
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> ans=new ArrayList();
if(nums.length==0||nums.length<4)
return ans;
if(nums.length==4&&(nums[0]+nums[1]+nums[2]+nums[3]==target))
{
ans.add(Arrays.asList(nums[0],nums[1], nums[2], nums[3]));
return ans;
}
Arrays.sort(nums);
int end=nums.length-1;
for(int i=0;i<nums.length-3;i++){
if(i>=1&&nums[i]==nums[i-1])
continue;
for(int j=i+1;j<nums.length-2;j++){
if(j>i+1&&nums[j]==nums[j-1])
continue;
int l=j+1;
int r=end;
while(l<r){
if(nums[i]+nums[l]+nums[r]+nums[j]>target)
r--;
else if(nums[i]+nums[l]+nums[r]+nums[j]<target)
l++;
else if(nums[i]+nums[l]+nums[r]+nums[j]==target){
// List<Integer>res=new ArrayList<Integer>();
// res.add(nums[i]);
// res.add(nums[l]);
// res.add(nums[r]);
while (l<r && nums[l] == nums[l+1]) l++; // 去重
while (l<r && nums[r] == nums[r-1]) r--;
ans.add(Arrays.asList(nums[i],nums[j], nums[l], nums[r]));
r--;
l++;
}
}
}
}
return ans;
}
}