题目
给定一个包含 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]
]
思路
-
要先对数组数组进行一个校验,判断是否为空或者数组长度小于4,如果是,直接返回空的集合。
-
遍历过程中需要使用到两级for循环,每一级循环开始时,都需要做以下判断
1、当前最小值 > target,则之后都不会满足,break;
2、当前最大值 < target,则当前循环不满足,进入下一个循环,continue; -
在每一级循环开始时,要做去重处理,判断当前值是否与之前的值相等且大于起始值,如果是,进入下一循环。
-
这个题目最关键的,其实就是对数组所有情况进行去重、过滤,只对最后少数符合条件的结果进行while操作,如果直接使用暴力解法肯定会有超时错误。
代码
public class Q018_fourSum {
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> ans = new ArrayList<List<Integer>>();
int len = nums.length;
if(nums == null || len < 4) return ans;
Arrays.sort(nums); // 排序
for (int i = 0; i < len - 3; i++) {
if(i > 0 && nums[i] == nums[i-1]) continue; // 去重
//最小值大于target
if((nums[i]+nums[i + 1]+nums[i + 2]+nums[i + 3]) > target) break;
//最大值小于target
if((nums[i] + nums[len - 1]+nums[len - 2]+nums[len - 3]) < target) continue;
for(int j = i + 1; j < len - 2; j++){
if(j > (i + 1) && nums[j] == nums[j-1]) continue; // 去重
int L = j+1;
int R = len-1;
//最小值大于target
if((nums[i] + nums[j] +nums[L] + nums[L + 1]) > target) break;
//最大值小于target
if((nums[i] + nums[j] +nums[R] + nums[R - 1]) < target) continue;
while(L < R){
int sum = nums[i] + nums[j] + nums[L] + nums[R];
if(sum == target){
ans.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 if (sum > target) R--;
}
}
}
return ans;
}