18. 四数之和
难度中等
标签:双指针,数组
给定一个包含 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]
]
思路:
1.先排序,从小到大
2.需要四个指针,i ,j分别我是第一个和第二指针,分别是两重循环
L,R分别是num[j]后面元素的集合中第一个元素和最后一个元素
3.nums[i]是最外层循环,如果4*nums[i]>target 结束循环,如果nums[i]和其上一个元素nums[i-1]相等跳过,注意i必须从第二个元素开始,即i>1
4.nums[j]是第一层循环,如果nums[i]+3nums[j]>target 结束这层循环,如果nums[j]和上一个元素nums[j-1]相等,注意j是从i后的第二个元素开始,即j>i+1
5.最内层使用双指针的思路,L,R分别是num[j]后面元素的集合中第一个元素和最后一个元素.如果nums[i]+nums[j]+nums[L]+nums[R]等于target,则生成一个List,装入四个元素,然后将List,添加到存放List的容器lists中,然后寻找下一组,L左移,R左移,需注意L,R左移时,要跳过下一个和他们相等的元素,去掉重复的
如果四数之和大于target, R左移,同样跳过和下一个元素相同的
如果四数之和小于target, L左移,同样跳过和下一个元素相同的
6.返回存放List的容器lists
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
if(nums.length<4) return null;
List<List<Integer>> lists= new ArrayList<>();
//先排序
Arrays.sort(nums);
int sum=0;
for(int i=0;i<nums.length-3;i++){
//去重
if(i>0&&nums[i]==nums[i-1]) continue;
//四个最小的大于目标必定没有这个的四个数了,结束循环
if(4*nums[i]>target) break;
for(int j=i+1;j<nums.length-2;j++){
//去重
if(j>i+1&&nums[j]==nums[j-1]) continue;
//这层已经没有这样的四个数,跳过这层循环,寻找下面的
if(nums[i]+3*nums[j]>target) break ;
//左指针
int L=j+1;
//右指针
int R=nums.length-1;
while(L<R){
sum=nums[i]+nums[j]+nums[L]+nums[R];
//相等时
if(sum==target){
//生成一个List放入四个数,将这个List放入结果中
lists.add(Arrays.asList(nums[i],nums[j],nums[L],nums[R]));
//去重
while(L<R&&nums[L]==nums[L+1])
L++;
while(L<R&&nums[R]==nums[R-1])
R--;
//两个指针左移
L++;
R--;
}
else if(sum>target){
//去重
while(L<R&&nums[R]==nums[R-1])
R--;
//R左移
R--;
}
else {
while(L<R&&nums[L]==nums[L+1])
L++;
L++;
}
}
}
}
return lists;
}
}