原题链接:Leecode 18. 四数之和
注意数据范围,这题和之前的Leecode 15. 三数之和 双指针的思路基本一致,都是排序+双指针,但是此题中nums
数组中数据较大,4数相加可能爆int
,要注意数据范围的判断。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> res;
if(nums.size()<4)
return res;
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size();i++)
{
if(i>0&& nums[i]==nums[i-1])//去重
continue;
for(int j=i+1;j<nums.size();j++)
{
if(j>i+1&& nums[j]==nums[j-1])//去重
continue;
int l=j+1,r=nums.size()-1;
while(l<r)
{
long long a=nums[i];long long b=nums[j];
long long c=nums[l];long long d=nums[r];
long long tmp=a+b+c+d;
if(tmp>INT_MAX)
return res;
if(tmp==target)
{
res.push_back({nums[i],nums[j],nums[l],nums[r]});
while(l<r && nums[r]==nums[r-1])//去重
r--;
while(l<r && nums[l]==nums[l+1])//去重
l++;
l++;r--;
}
else if(tmp>target)
r--;
else if(tmp<target)
l++;
}
}
}
return res;
}
};