题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
给定一个包含 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]
]
方法是二重for循环加上双指针法
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
Arrays.sort(nums);
List<List<Integer>> rs = new ArrayList<>();
int len = nums.length;
for (int i=0;i<len;i++){
for (int j=i+1;j<len;j++){
int left = j+1,right = len-1;
while (left<right){
int sum = nums[i]+nums[j]+nums[left]+nums[right];
if (sum<target){
while (left<right&&nums[left]==nums[++left]);
}else if (sum>target){
while (left<right&&nums[right]==nums[--right]);
}else {
rs.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]);
}
}
}
}
//利用HashSet去除重复元素
HashSet hashSet = new HashSet(rs);
rs.clear();
rs.addAll(hashSet);
return rs;
}
}
上面的用到了HashSet,虽然能达到目的且能成功AC,但是可以进一步优化,在for循环里面就可以进行去除重复元素,减少运行时间和空间
如下:
public List<List<Integer>> fourSum(int[] nums, int target) {
Arrays.sort(nums);
List<List<Integer>> rs = new ArrayList<>();
if (nums==null||nums.length<4){
return rs;
}
int len = nums.length;
//第一层for循环不遍历全部,因为至少要留三个数给后面的,不然不符合题意
for (int i=0;i<len-3;i++){
//跳过重复元素,避免出现重复结果
if (i>0&&nums[i]==nums[i-1]) continue;
//同上,至少要留两数给后面的
for (int j=i+1;j<len-2;j++){
//跳过重复元素,避免出现重复结果
if (j>i+1&&nums[j]==nums[j-1]) continue;
int left = j+1,right = len-1;
//双指针
while (left<right){
int sum = nums[i]+nums[j]+nums[left]+nums[right];
if (sum<target){
//sum比目标小,左指针右移
//同时跳过重复元素,避免出现重复结果
while (left<right&&nums[left]==nums[++left]);
}else if (sum>target){
//sum比目标大,右指针左移
//同时跳过重复元素,避免出现重复结果
while (left<right&&nums[right]==nums[--right]);
}else {
//sum和target相等就加入到结果列表
rs.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]);
}
}
}
}
return rs;
}
两次执行的结果对比