题目描述:
给定一个包含 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]
]
分析:
这题和前面的一道三数之和
类似,解题的思路都一样,这里直接选取两个基准就可以了,然后循环出所有的组合进行判断,如果正好相等那么就加入Set集合中。
代码如下:
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
// 定义一个Set集合用来存放结果集
Set<List<Integer>> set = new HashSet<>();
// 如果为空直接返回
if(nums.length == 0) {
return new ArrayList<>(set);
}
// 对数组进行排序处理
Arrays.sort(nums);
int length = nums.length;
// 因为是四个数比较,所以第一个基准循环到倒数第四个元素即可
for (int i = 0; i < length - 3; i++) {
int temp1 = nums[i];
// 同样的基准二
for (int j = i + 1; j < length - 2; j++) {
int temp2 = nums[j];
// 同样的左右两个指针
int left = j + 1;
int right = length - 1;
while (left < right) {
// sum 等于4个数相加
int sum = temp1 + temp2 + nums[left] + nums[right];
// 如果小于targer,做指针右移,大于target,右指针左移,等于就添加到Set集合中
if (sum < target) {
left++;
} else if (sum > target) {
right--;
} else {
set.add(Arrays.asList(temp1, temp2, nums[left++], nums[right--]));
}
}
}
}
return new ArrayList<>(set);
}
}
总结:
时间复杂度为 O ( n 3 ) O(n^3) O(n3),思路还是挺好理解的,但效率略慢,需要3层for循环。