四个数无非就是加一层for循环,值得注意的是,四个数中,第一个数、第三个数、第四个数的去重处理可以直接照搬上面的代码,加一下第二个数的去重处理即可。
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> res=new ArrayList<List<Integer>>();
if (nums.length < 4) {
return res;
}
Arrays.sort(nums);
for(int i=0;i<nums.length;i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
for (int j = i + 1; j < nums.length; j++) {
if(j > i + 1 && nums[j] == nums[j - 1]) {
continue;
}
int l = j + 1;
int r=nums.length-1;
while(l<r) {
int sum=nums[i]+nums[j]+nums[l]+nums[r];
if (sum == target) {
res.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) {
++l;
}else {
--r;
}
}
}
}
return res;
}