题目
给定一个包含 n 个整数的数组 nums
和一个目标值 target
,判断 nums
中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target
相等?找出所有满足条件且不重复的四元组。
思路
四数之和与之前的三数之和、两数之和存在共通性,即:都要确定左边界和右边界
在此题中,首先设置第一个数a在数组的最左端,第二个数b在a+1位置,第三个数left在b+1位置,第四个数right在nums.length-1的位置。
若left小于right,通过left和right相互逼近获取符合条件的数组,在获取到数组后,再判断nums[left] == nums[left+1]
判断当前元素与下一个元素的值是否相等,相等需要跳过,最后返回整个结果集。
解题
public List<List<Integer>> fourSum(int[] nums, int target) {
// 判断是否是合法数组
int size = nums.length;
if (size < 4) {
return Collections.emptyList();
}
// 排序
Arrays.sort(nums);
// 定义List接受结果
List<List<Integer>> lists = new ArrayList<>();
// 最外层循环
for (int a = 0; a < size; a++) {
// 如果最左边的数已经向右移动了,就判断和之前的数是否一样,若一样就跳出此次循环(获取第一个数)
if (a > 0 && nums[a] == nums[a - 1]) {
continue;
}
// 第二个数是在第一个数的基础上加1(获取第二个数)
for (int b = a + 1; b < size; b++) {
if (b > a + 1 && nums[b] == nums[b - 1]) {
continue;
}
// 定义第三个数(获取第三个数)
int left = b + 1;
// 把最后一个数放在数组的尾端(获取第四个数)
int right = size - 1;
// 若左指正小于右指针,执行循环
while (left < right) {
// 若四数之和小于target,执行left++ 使target变大
if (nums[a] + nums[b] + nums[left] + nums[right] < target) {
left++;
// 若四叔之和大于target,执行right++,是target变小
} else if (nums[a] + nums[b] + nums[left] + nums[right] > target) {
right--;
} else {
// 四叔之和等于target 放入list中
lists.add(Arrays.asList(nums[a], nums[b], nums[left], nums[right]));
// 判断left将要右移的数是不是等于当前的值,若等于,执行left++
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
// 判断right将要右移的数是不是等于当前的值,若等于,执行right--
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
// left++ 继续执行下一次探索
left++;
// left++ 继续执行下一次探索
right--;
}
}
}
}
// 返回最后结果
return lists;
}
总结
有了两数之和和三数之和的经验,解决此题的思路就比较容易,但在自己写代码的过程中,还是会出现bug,还是一些小细节没有考虑周全