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)
在解题时候上面的报错给我整笑了,可谓是奇奇怪怪。好吧那处了问题还是解决一下,在本题中有两种解决方案
- 因为本题中的基数其实不是很大,所以我们可以采取预先计算后两重循环需要计算的值再比较,也就是如代码中所示的在后面的双指针过程中直接判读
nums[left]+nums[right]
与temp
的关系而不是四个值全部加起来。 - 或者就是使用更大范围的数据类型一次性解决。
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;
}
};