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
  你可以按 任意顺序 返回答案 。

示例1：

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

示例2：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/4sum

2.题目分析

题目与前面的三数之和很相似，解法也相似。不同的就是三数之和是由三重循环优化到两重循环，而本题是由四层循环优化到三层循环。基本思路就是先确定其中的两个数，然后采用双指针进行其他两个数的选择。如果做过前面的三数之和的话，对本题的思路就会有着很清晰的了解。

不过解题时也有一些细节需要把握。我们假设从左满足条件的四个数从左到右的下标依次是i,j,k,l。那么我们先确定哪两个下标就决定了后续解题的编码实现的方向。我在解题时想着先以确定[i,l]为边界的区间然后再在其中寻找适合的j和k。这样就可以将循环降低到两重，但是事实是j和k很容易就可以进行确定下标与去重等操作，但是之后对于[i,l]的更新却成了一个让人头疼的问题，我暂时是没有想出来怎么更新才合理。

辗转再三，于是采用先确定左边两个下标i和j，然后再循环确定右边两个下标k和l，这样的话代码逻辑会更加清晰。同三数之和一样，本体还是要进行繁琐的去重操作。具体代码见第三部分。

另外就是，在三数之和的解题中我们会根据排序后的第一个元素是否大于0决定是否直接结束循环，但是在本题中的任意target的条件下是不能这样判断的。看下面一个例子，此时就不能因为排序后的-5>-11而直接跳出循环。

runtime error: signed integer overflow: 2000000000 + 1000000000 cannot be represented in type ‘int’ (solution.cpp)

在解题时候上面的报错给我整笑了，可谓是奇奇怪怪。好吧那处了问题还是解决一下，在本题中有两种解决方案

1. 因为本题中的基数其实不是很大，所以我们可以采取预先计算后两重循环需要计算的值再比较，也就是如代码中所示的在后面的双指针过程中直接判读nums[left]+nums[right] 与temp的关系而不是四个值全部加起来。
2. 或者就是使用更大范围的数据类型一次性解决。

3.题目解答

题目代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        //初始化结果向量
        vector<vector<int>> result;
        //预存数组长度
        int len = nums.size();
        //数组排序以方便处理
        sort(nums.begin(),nums.end());
        //定义循环左边界i
        for(int i=0;i<len;i++){
            if(i>0 && nums[i]==nums[i-1])continue;
            //定义循环左边第二个数jj
            for(int j=i+1;j<len;j++){
                if(j>i+1 && nums[j]==nums[j-1])continue;
                //转换为求left和right的和的子问题
                //还可以防止溢出
                int temp = target-nums[i]-nums[j];
                int left = j+1,right=len-1;
                //确定最左边两个数后求其他两个数
                while(left<right){
                    if(nums[left]+nums[right] ==temp){
                        result.push_back({nums[i],nums[j],nums[left],nums[right]});
                        //左边去重
                        while(left<right && nums[left]==nums[left+1])left++;
                        //右边去重
                        while(left<right && nums[right]==nums[right-1])right--;
                        right--;
                        left++;
                    }else if(nums[left]+nums[right] > temp) right--;
                    else left++;
                }
            }
        }
        return result;
    }
};