和3sum一样的意思,给一个序列和一个目标,在序列中找出不重复的四个数的和为目标的所有情况。
思路和3sum也一样,只是需要先将两个数相加,转化为3sum中的一个数,其他的就用3sum来做。
PS:开始是这样想的,但是优化没有这么变态,看了下discuss中的优化,简直了,leetcode中的优化感觉都是在很小很小的角落里优化的那种感觉,而且没有给定规定的时间复杂度,所以只能慢慢试了= =。。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int> >w;
vector<int> temp;
if(nums.size()<4)return w;
temp.resize(4);
sort(nums.begin(),nums.end());
for(int ii=0;ii<nums.size()-3;ii++){
if(nums[ii]+nums[ii+1]+nums[ii+2]+nums[ii+3]>target)break;//优化
if(ii&&(nums[ii]==nums[ii-1]))continue;
for(int jj=ii+1;jj<nums.size()-2;jj++){
if(nums[ii]+nums[jj]+nums[jj+1]+nums[jj+2]>target)break;//优化
if((jj>ii+1)&&nums[jj]==nums[jj-1])continue;
int r=nums.size()-1;
int l=jj+1;
int ans=target-(nums[ii]+nums[jj]);
while(l<r){
if(nums[l]+nums[r]==ans){
temp[0]=nums[ii];
temp[1]=nums[jj];
temp[2]=nums[l];
temp[3]=nums[r];
w.push_back(temp);
l++;
r--;
while(l<nums.size()&&nums[l]==nums[l-1])l++;//检查重复的
while(r>-1&&nums[r]==nums[r+1])r--;//检查重复的
}
while(nums[l]+nums[r]>ans)r--;
while(nums[l]+nums[r]<ans)l++;
}
}
}
return w;
}
};