Leetcode 18四数之和（双指针+简单剪枝）

解题思路：

大暴力+双指针+简单剪枝。（见代码）

AC代码：

 

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int> > ans;
        int len = nums.size();
        if(len < 4) return ans;
        sort(nums.begin(), nums.end());    //排序
        for(int i = 0; i < len - 3; i++) {
            //剪枝1 当四个数字大于target的时候可以结束循环
            if(nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) break;
            //剪枝2 如果nums中两数相等则可直接跳过以避免重复
            if(i > 0 && nums[i] == nums[i - 1]) continue;
            for(int j = i + 1; j < len - 2; j++) {
                //同剪枝1 但是注意此时nums[i]和nums[j]是确定的
                if(nums[j] + nums[j + 1] + nums[j + 2] + nums[i] > target) break;
                //同剪枝2
                if(j > i + 1 && nums[j] == nums[j - 1]) continue;
                //左右指针
                int l = j + 1, r = len - 1;
                while(l < r) { 
                    int sum = nums[i] + nums[j] + nums[l] + nums[r];
                    if(sum == target) {
                        ans.push_back({nums[i], nums[j], nums[l], nums[r]});
                        while(l < r && nums[l] == nums[l + 1]) {
                            l++;
                        }
                        //加完
                        l++;
                        while(l < r && nums[r] == nums[r - 1]) {
                            r--;
                        }
                        //减完
                        r--;
                    }
                    else if(sum < target) l++;
                    else r--;
                } 
            }
        }
        return ans;
    }
};