1.题目解析
题目链接:https://leetcode.cn/problems/4sum/submissions/
给你一个由 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]]
2.算法原理
思想:排序 + 双指针
1.依次固定一个数a。
2.在a后面的区间内,利用“三数之和”的思想,找到三个数,使这三个数的和等于target - a
即可。
1)依次固定一个数 b。
2)在b后面的区间内,利用 “双指针” 找到两个数,使这两个数的和等于target - a - b
即可。
3.处理细节问题
1)不重
2)不漏
处理细节的步骤几乎和处理三数之和一模一样,这里就不在讲解了,大家可以去看我的另一篇博客:三数之和
3.代码演示
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target)
{
vector<vector<int>> ret;
// 1.排序
sort(nums.begin(), nums.end());
// 2.利用双指针解题
int n = nums.size();
for(int i = 0; i < n; i++)
{
for(int j = i + 1; j < n; j++)
{
int left = j + 1;
int right = n - 1;
long long t = (long long)target -(nums[i] + nums[j]);
while(left < right)
{
int sum = nums[left] + nums[right];
if(sum > t) right--;
else if(sum < t) left++;
else
{
ret.push_back({nums[i], nums[j], nums[left], nums[right]});
left++, right--;
// 去重
while(left < right && nums[left] == nums[left - 1]) left++;
while(left < right && nums[right] == nums[right + 1]) right--;
}
}
// j去重
while(j < n - 1 && nums[j] == nums[j + 1]) j++;
}
// i去重
while(i < n - 1 && nums[i] == nums[i + 1]) i++;
}
return ret;
}
};
注意:在leedcode中,本题long long t = (long long)target -(nums[i] + nums[j]);
这段代码,如果不使用long long
,会有数据溢出的风险,因为有一组很大的测试数据,范围超过了int
。