题目描述:
给定一个包含 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 ) ,思路还是挺好理解的,但效率略慢,需要3层for循环。