class Solution {
public:
void threeSum(vector<int>nums, long target, int start, vector<vector<int> >&temp)
{
for(int i=start; i<(int)nums.size()-2;)
{
int left = i + 1;
int right = nums.size()-1;
while(left<right)
{
long sum = (long)nums[i] + (long)nums[left] + (long)nums[right];
if(sum==target)
temp.push_back({nums[i], nums[left], nums[right]});
if(sum<target)///<这里不能用else if,相等的时候也得加
{
do
{
++left;
}while(left<right && nums[left-1]==nums[left]);
}
else
{
do
{
--right;
}while(right>left && nums[right]==nums[right+1]);
}
}
do
{
++i;
}while(i<(int)nums.size() && nums[i-1]==nums[i]);
}
}
vector<vector<int>> fourSum(vector<int> nums, int target) {
vector<vector<int> >ret;
sort(nums.begin(), nums.end());
for(int i=0; i<(int)nums.size()-3;)
{
vector<vector<int> >temp;
threeSum(nums, target-nums[i], i+1, temp);
for(auto vec: temp)
{
vec.push_back(nums[i]);
ret.push_back(vec);
}
do
{
++i;
}while(i<(int)nums.size() && nums[i-1]==nums[i]);
}
return ret;
}
};
18. 四数之和
最新推荐文章于 2023-02-09 12:02:20 发布