问题描述:给定一个包含 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]]
思路:
- 暴力法
遍历所有情况找出所有满足情况的数组,用先对满足的数字序列排序来达到去重的目的。
时间复杂度:o(n4)
空间复杂度:o(n)
- 排序+指针
与三数之和类似,先排序,对前两个元素进行遍历,在遍历过程中设置后续两个左右指针,通过与target的比较情况来进行移动。
时间复杂度:o(n3)
空间复杂度:o(1)
//java
//暴力法
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
if(nums.length < 4) return new ArrayList();
Arrays.sort(nums);//先排序,方便后续去重
List<List<Integer>> list = new ArrayList<>();
for(int i = 0; i < nums.length; ++i){
int targeti = target - nums[i];
for(int j = i + 1; j < nums.length; ++j){
for(int k = j + 1; k < nums.length; ++k){
for(int p = k + 1; p < nums.length; ++p){
if(nums[j] + nums[k] + nums[p] == targeti){
List<Integer> ls = new ArrayList<Integer>(Arrays.asList(nums[i], nums[j], nums[k], nums[p]));
if(!list.contains(ls)) list.add(ls);//去重
}
}
}
}
}
return list;
}
}
//排序+指针
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
if(nums.length < 4) return new ArrayList();
Arrays.sort(nums);
List<List<Integer>> list = new ArrayList<>();
int len = nums.length;
for(int i = 0; i < len-3; ++i){
if(i > 0 && nums[i] == nums[i-1]) continue;//相同元素则跳过,去重
//优化,一旦出现后续遍历不可能满足的情况,立即剪枝
int min = nums[i] + nums[i+1] + nums[i+2] + nums[i+3];
if(min > target) break;//最小情况大于目标值,跳过后续所有遍历
int max = nums[i] + nums[len-1]+nums[len-2]+nums[len-3];
if(max < target) continue;//最大情况小于目标值,在该位置上的后续指针移动不可能满足,跳过当前位置
for(int j = i + 1; j < len-2; ++j){
if(j > i+1 && nums[j] == nums[j-1]) continue;//相同元素则跳过
int left = j + 1;
int right = nums.length - 1;
//优化
int min1 = nums[i] + nums[j] + nums[j+1] + nums[j+2];
if(min1 > target) break;
int max1 = nums[i] + nums[j]+nums[right]+nums[right-1];
if(max1 < target) continue;
while(left < right){
int sum = nums[i] + nums[j] + nums[left] + nums[right];
if(sum == target){
list.add(new ArrayList<Integer>(Arrays.asList(nums[i],nums[j], nums[left], nums[right])));
//跳过重复值
while(left < right && nums[left] == nums[++left]){}
while(left < right && nums[right] == nums[--right]){}
}else if(sum > target)
--right;
else
++left;
}
}
}
return list;
}
}