在三数之和的基础上理解这题的设计思路不难,个人认为难点在于去重的处理。
下面以数组[0,0,0,0,0,1,2,3,3,3]举例说明:
- 第一层遍历的去重:当数字第一次等于0时,此时不应该判断去重,因为题目没有限制后面的三个数字不能重复,如果第一次等于0时去重,则后面的数字不能取到0。所以
nums[i] == nums[i-1]
的判断条件保证了在一连串的重复数字中,第一次取到该数字时不会去重。 - 第二层遍历同理,后面两个数字可以与第二个数字重复,第一次取某值时不应去重。
- 最后两个数字的去重,比较简单,用while循环即可实现。
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
for(int i=0; i<nums.length-3;i++){
if(i>0 && nums[i] == nums[i-1]){continue;}
int target1 = target - nums[i];
for(int j=i+1; j<nums.length-2;j++){
if(j>i+1 && nums[j] == nums[j-1]){continue;}
int target2 = target1 - nums[j];
int k=j+1;
int l = nums.length-1;
while(k<l){
if(nums[k] + nums[l] == target2){
res.add(Arrays.asList(nums[i], nums[j], nums[k], nums[l]));
while(k<l && nums[k] == nums[k+1])k++;
while(k<l && nums[l] == nums[l-1])l--;
k++;
l--;
}else if(nums[k] + nums[l] < target2){
k++;
}else{
l--;
}
}
}
}
return res;
}
}