一,题目
给你一个由 n
个整数组成的数组 nums
,和一个目标值 target
。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]]
(若两个四元组元素一一对应,则认为两个四元组重复):
0 <= a, b, c, d < n
a
、b
、c
和d
互不相同nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。
示例 1:
输入:nums = [1,0,-1,0,-2,2], target = 0 输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2:
输入:nums = [2,2,2,2,2], target = 8 输出:[[2,2,2,2]]
二,解
本题依然采用双指针法
具体来说,对比三数之和最快的双指针方法,我们只需在外层在嵌套一层循环即可。这样时间复杂度将从O(N^2)变为 O(N^3)
提升速度方法(判重,边界条件):
指针依次是 p,k,i,j,如果 nums[p] + 3 * nums[p + 1] > target,因为 nums 按升序排列,所以之后的数肯定都大于 target ,直接 break;
如果 nums[p] + 3 * nums[-1] < target,那么当前的 nums[p] 加其余三个数一定小于 target,故 p 直接下一位即可,continue;
k 和 p 判断完全一样,只是将 3 变成了 2,target 变成了 target - nums[p]。
同样地,为了避免结果重复,某个指针遇到相同的数需要直接跳过,这与三数之和是一样的。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> res;
if(nums.size()<4)return res;
size_t n=nums.size();
sort(nums.begin(),nums.end());
for(int i=0;i<n-3;i++){
//边界条件,不存在
if(target<=0&&nums[i]>0)break;
if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target)break;
//条件不满足
if(nums[i]+nums[n-3]+nums[n-2]+nums[n-1]<target)continue;
//去重
if(i>0&&nums[i]==nums[i-1])continue;
for(int j=i+1;j<n-2;j++){
//边界条件,不存在
if(nums[i]+nums[j]+nums[j+1]+nums[j+2]>target)break;
//条件不满足
if(nums[i]+nums[j]+nums[n-2]+nums[n-1]<target)continue;
//去重
if(j>i+1&&nums[j]==nums[j-1])continue;
int left=j+1,right=n-1;
while(left<right){
//a:nums[i],b:nums[j],c:nums[left],nums[j],求当前四元组之和,和目标target比较
int sum=nums[i]+nums[j]+nums[left]+nums[right];
if(sum<target)left++;
//双指针移动
else if(sum>target)right--;
else {
//找到对应四元组
res.push_back({nums[i],nums[j],nums[left],nums[right]});
int last_start=left,last_end=right;
//找到四元组后去重c,d
while(left<right&&nums[left]==nums[last_start])left++;
while(left<right&&nums[right]==nums[last_end])right--;
}
}
}
}
return res;
}
};