链接:戳这里
题意:给出长度为n的整数序列和目标target,定义一个四元组a[i]+a[j]+a[k]+a[l]==target
输出所有满足条件的四元组 ,两个四元组不同当且仅当至少一个元素不同
思路:先升序排序,枚举当前a[i],然后再枚举a[j],接下来在区间[j+1,n-1]中找出两个a[l],a[r]使得a[i]+a[j]+a[l]+a[r]=tar
因为是升序排序,a[i]+a[j]=定值=sum,所以当:
1:a[l]+a[r]+sum<tar的时候 l++
2:a[l]+a[r]+sum>tar的时候 r--
3:a[l]+a[r]+sum=tar 统计答案
这里的去重需要扣一下细节(就是如果当前这个位置pos如果作为相同的值在之前已经统计过了,就不需要统计了),当然啦你也可以直接unique
具体看代码O(n^3)
代码:
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
int n=nums.size();
vector<vector<int> > ans;
if(n<4) return ans;
sort(nums.begin(),nums.end());
for(int i=0;i<n-3;i++){
if(i>0 && nums[i]==nums[i-1]) continue;
if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target) break;
if(nums[n-1]+nums[n-2]+nums[n-3]+nums[i]<target) continue;
for(int j=i+1;j<n-2;j++){
if(j>i+1 && nums[j]==nums[j-1]) continue;
if(nums[i]+nums[j]+nums[j+1]+nums[j+2]>target) break;
if(nums[n-1]+nums[n-2]+nums[j]+nums[i]<target) continue;
int l=j+1,r=n-1;
while(l<r){
int sum=nums[i]+nums[j]+nums[l]+nums[r];
if(sum<target || (l>j+1 && nums[l]==nums[l-1])) l++;
else if(sum>target || (r+1<n-1 && nums[r]==nums[r+1])) r--;
else ans.push_back({nums[i],nums[j],nums[l++],nums[r--]});
}
}
}
return ans;
}
};