题目
给定一个包含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]
]
解题思路
- 当时做题18的时候,先是做了1-两数之和,15-三树之和,454-四数相加Ⅱ三道题。从两数之和到三数之和当时用的思想就是化简,将三数之和的问题化简到两数之和问题,然后套用经典的双指针就可以解决了。同理本题的四数之和同样可以沿着化简的思路,一路化简到三数之和,两数之和。
- 下面的话,直接用代码实现一下,主要要注意的地方就是边界情况以及如何去避免冗余计算(考虑这一点可以很大的优化时间复杂度)。代码是可以AC的。
代码
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> ret = new ArrayList<>();
//特殊情况的检查
if(nums==null||nums.length<4){
return ret;
}
Arrays.sort(nums);
int len = nums.length;
//i,j,start,end分别是四个指针
for(int i = 0; i < len - 3 ; i++){
//去重,如果当前元素和上一个元素相等的话,因为上一个元素已经考虑过了,所以当前元素就不需要考虑了。
if( i > 0 && nums[i] == nums[i-1]){
continue;
}
//避免冗余计算,主要是在两种情况下可以不继续往后计算了。
//1:当前元素的最小四数之和如果都大于target的话,显然后面不可能有等于target的组合,且当前元素已经不能再变大,直接退出循环。
//2:当前元素的最大四数之和如果都小于target的话,当前元素也就不会存在满足的组合,当前元素需要后移一位。
int minValue = nums[i] + nums[i+1] + nums[i+2] + nums[i+3];
if(minValue > target)
break;
int maxValue = nums[i] + nums[len-3] + nums[len-2] + nums[len - 1];
if(maxValue < target)
continue;//此时实现了i++,即右移。
//上面一重for循环已经将四数之和化简到了三数之和,现在化简到两数之和。
for(int j = i + 1; j < len -2 ; j++){
//同理,去重
if(j > i+1 && nums[j] == nums[j-1])
continue;
//继续避免冗余计算,思路同上
int minValue2 = nums[i] + nums[j] + nums[j+1] + nums[j+2];
int maxValue2 = nums[i] + nums[j] + nums[len-2] + nums[len-1];
if(minValue2 > target)
break;//如果此时j和i之间有间隔的话,i->i+1,j->i+2,这种情况下可能还有解
if(maxValue2 < target)
continue;//j++,右移
//此时就已经化简到了两数之和的部分了
int start = j + 1, end = len -1;
while(start < end){
int curValue = nums[i] + nums[j] + nums[start] + nums[end];
if(curValue == target){
ret.add(Arrays.asList(nums[i],nums[j],nums[start],nums[end]));
//维护start和end指针,先是去重(找到和当前元素不一样的下个元素)然后以动
while(start < end && nums[start] == nums[start+1]) start++;
while(start < end && nums[end] == nums[end-1]) end--;
start++;
end--;
}
else if( curValue > target)
end--;
else
start++;
}
}
}
return ret;
}