首先对数组进行排序,为后续的去重做准备。
四数之和可以简化理解:
(1)首先固定数组的左右两端(想象成窗口两端,i表示窗口左端,j表示窗口右端),用双指针(一个指针left从左往右,一个指针right从右往左)在窗口内进行循环,直到两个指针相遇或者左指针在右指针的右侧或右指针在左指针的左侧时结束窗口循环(第一种情况是因为题目要求不能有位置相同的元素,剩下两种情况是因为指针相错后其实进行的循环就重合了)。
(2)在窗口两端固定时,遍历窗口内的组合形式(不重复的情况下),当找到符合条件时,保留此时的四个元素,并对left指针right指针进行去重操作
(3)在窗口(left right)遍历一轮结束后,移动窗口右端指针j,并重复上述操作
(4)窗口右端遍历后,再移动窗口左端遍历,重复上述操作
代码如下:
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> list = new ArrayList<List<Integer>>();
Arrays.sort(nums);
//可以继续进行剪枝操作进行优化
for(int i =0; i < nums.length; i++){
if(i > 0 && nums[i] == nums[i - 1]){
continue;
}
for(int j = nums.length - 1; j > i && (j-i) >=3; j--){
if(j < nums.length - 1 && nums[j] == nums[j + 1]){
continue;
}
int left = i + 1;
int right = j - 1;
while(right > left){
long sum = (long) nums[i] + nums[j] + nums[left] + nums[right];
if(sum > target){
right--;
}else if(sum < target){
left++;
}else{
List<Integer> tem = new ArrayList<Integer>();
tem.add(nums[i]);
tem.add(nums[j]);
tem.add(nums[left]);
tem.add(nums[right]);
list.add(tem);
while(right > left && nums[right] == nums[right - 1]){
right --;
}
while(right > left && nums[left] == nums[left + 1]){
left++;
}
right--;
left++;
}
}
}
}
return list;
}
}