tags: hash table、two pointers
k-sum问题都是一类问题。
最经典的是2-sum,用two pointers在线性时间内解决。
至于3sum、4sum问题,都可以先将问题分解,然后在用2sum的方法解决。2sum时间复杂度O(N^2),3sum时间复杂度O(N^3)。
具体见:http://www.sigmainfy.com/blog/summary-of-ksum-problems.html
代码:
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
vector<vector<int>> res;
vector<int> temp;
int size = nums.size(), i, j, p1, p2;
for(i = 0; i < size-1; ++i){
if(i>0&&nums[i]==nums[i-1]) continue;
for(j = i+1; j < size; ++j){
if(j>i+1&&nums[j]==nums[j-1]) continue;
temp.clear();
int t = target - nums[i] - nums[j];
temp.push_back(nums[i]);
temp.push_back(nums[j]);
p1 = j+1;
p2 = size-1;
while(p1<p2){
if(p1>j+1&&nums[p1]==nums[p1-1]){
p1++;
continue;
}
if(nums[p1]+nums[p2]==t){
temp.push_back(nums[p1]);
temp.push_back(nums[p2]);
res.push_back(temp);
temp.pop_back();
temp.pop_back();
p1++;p2--;
}
else if(nums[p1]+nums[p2]<t) p1++;
else p2--;
}
}
}
return res;
}
};
提交后一看128ms,前面的都是12ms、16ms。查了后发现别人多了剪枝。剪枝很强大啊!!!
代码:
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> total;
int n = nums.size();
if(n<4) return total;
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[i]+nums[n-3]+nums[n-2]+nums[n-1]<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[i]+nums[j]+nums[n-2]+nums[n-1]<target) continue;
int left=j+1,right=n-1;
while(left<right){
int sum=nums[left]+nums[right]+nums[i]+nums[j];
if(sum<target) left++;
else if(sum>target) right--;
else{
total.push_back(vector<int>{nums[i],nums[j],nums[left],nums[right]});
do{left++;}while(nums[left]==nums[left-1]&&left<right);
do{right--;}while(nums[right]==nums[right+1]&&left<right);
}
}
}
}
return total;
}
};
两种代码,结构一样的,都是两个for循环(寻找两个数),再加上2sum的two pointers的算法(再找另外两个数)。
不一样的地方在:
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(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;
至于hash table的方法,4SUM时间能降到O(N^2)。
详细见:http://www.sigmainfy.com/blog/summary-of-ksum-problems.html
不过我还没有实现。12ms/16ms 已经是最好的了,这种方法应该不会超过12/16吧?
有时间再实现下。