leetcode 18.四数之和

1.题目

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。

2.示例

1）示例 1：
输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

2）示例 2：
输入：nums = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

3.数据范围

1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

4.分析

这道题目是三数之和的进阶版，多了一个数，将原本的单层循环改为外围双层循环枚举前两个数a,b即可，且四个数满足a<b<c<d，在此基础上需要在意几个细节，第一，为了满足四元组不重复，需要特殊判断四个节点，对于第一个节点a，只要当前节点和前一个节点相同，双明已经枚举了a作为第一小的情况了，对于第二个节点b，要满足j>a+1且和前一个点相同，这样说明也已经枚举了b作为第二小的情况了，c,d排除重复同理。在此基础上，需要注意tarhet的范围在计算a+b+c+d时候可能会出现超出int的范畴，且虽然定义了longlong sum=a+b+c+d，但计算时仍然是先存在int类型中的，会报错：runtime error: signed integer overflow，可以先把a,b,c,d都转化为long long 形式的就可以解决问题了。

5. 代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ans;
        if(nums.size()<4) return ans;
        sort(nums.begin(),nums.end());
        // ans.push_back(nums);
        for(int i=0;i<nums.size()-3;i++){
            if(i>0){
                if(nums[i]==nums[i-1]) continue;
            }
            for(int j=i+1;j<nums.size()-2;j++){
                if(j>i+1&&nums[j]==nums[j-1]) continue;
                int left=j+1,right=nums.size()-1;
                while(left<right){
                    long long a=nums[i],b=nums[j],c=nums[left],d=nums[right];
                    long long sum=a+b+c+d;
                    if(sum==target){
                        vector<int> test;
                        test.push_back(nums[i]);
                        test.push_back(nums[j]);
                        test.push_back(nums[left]);
                        test.push_back(nums[right]);
                        ans.push_back(test);
                        while(left<right&&nums[left]==nums[left+1]) left++;
                        while(left<right&&nums[right]==nums[right-1]) right--;
                    } 
                    if(sum>target) right--;
                    else left++;
                }
            }
        }
        return ans;
    }
};