四数之和

给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。

注意:

答案中不可以包含重复的四元组。

示例:

给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

满足要求的四元组集合为:
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

思路:需要定义四个下标变量:i,j,k,l,其中固定i,j,让k和l一个自增,一个自减。同样需要注意去重复,并且保证每个四元组按升序排列。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int> > vs;
        sort(nums.begin(),nums.end());                   
        for(int i=0; i<nums.size(); ++i){
            if (i>0 && nums[i-1]==nums[i]) continue;       
            for(int j=i+1; j<nums.size(); ++j){
                if (j>i+1 && nums[j-1]==nums[j]) continue;
                for(int k=j+1, l=nums.size()-1; k<l;){
                    if (k>j+1 && nums[k-1]==nums[k]){
                        ++k;                        
                        continue;
                    }
                    if (l<nums.size()-1 && nums[l]==nums[l+1]){
                        --l;                        
                        continue;
                    }
                    int sum = nums[i]+nums[j]+nums[k]+nums[l];
                    if (sum > target) --l;
                    else if (sum < target) ++k;
                    else {
                        vector<int> v(4,nums[i]);
                        v[1]=nums[j], v[2]=nums[k++], v[3]=nums[l--];
                        vs.push_back(v);
                    }
                }
            }
        }
        return vs;
    }
};

猜你喜欢

转载自blog.csdn.net/Aspiration_1314/article/details/88374087