4Sum

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吧？

有时间再实现下。