问题
给定一个包含 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]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析:
本题和三数之和相类似,三数之和是一个for循环固定一个指针,一个while(left<right)确定另外两个指针
而四数之和是两个for循环固定两个指针,while(left<right)确定两个指针
代码:
public class leet18 {
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> ans = new ArrayList<List<Integer>>();
int len = nums.length;
Arrays.sort(nums);
for(int i=0;i<len-3;i++)//-3:i后面有j、left、right预留出三个位置
{
if(i>0&&nums[i]==nums[i-1]) continue;//一次去重,如果i和前面的一样,直接continue
if(nums[i]+nums[len-1]+nums[len-2]+nums[len-3]<target) continue;//二次去复杂:如果当前i的最大都比target小则直接进入下一个i的循环
if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target) break;//去复杂:如果当前i最小都比target大,则直接结束,因为后面的只会更大
for(int j=i+1;j<len-2;j++)//j循环
{
if(j-i>1&&nums[j]==nums[j-1]) continue;//与i相同的去重去复杂
if(nums[i]+nums[j]+nums[len-1]+nums[len-2]<target) continue;
if(nums[i]+nums[j]+nums[j+1]+nums[j+2]>target) break;
int left=j+1;
int right=len-1;
while(left<right)
{
//这种方式行不通的原因:没有达到去重的目的
// if(nums[i]+nums[j]+nums[left]+nums[right]<target)
// {
// left++;
// }
// if(nums[i]+nums[j]+nums[left]+nums[right]>target)
// {
// right--;
// }
// if(nums[i]+nums[j]+nums[left]+nums[right]==target)
// {
// List<Integer> now = new ArrayList<Integer>();
// now.add(i);
// now.add(j);
// now.add(left);
// now.add(right);
// ans.add(now);
// continue;
// }
int temp = nums[i]+nums[j]+nums[left]+nums[right];
if(temp==target)
{
List<Integer> now = new ArrayList<Integer>();
now.add(nums[i]);
now.add(nums[j]);
now.add(nums[left]);
now.add(nums[right]);
ans.add(now);
while(left<right&&nums[left]==nums[left+1]) left+=1;//如果成功的情况下才去重,不成功不用去重
while(left<right&&nums[right]==nums[right-1]) right-=1;
//*
left++;
right--;
}else if(temp<target) {
left++;
}else {
right--;
}
}
}
}
return ans;
}
}