1. 题目解析
题目地址:四数之和
2. 讲解算法原理
- 首先对输入的数组进行排序,以便后续使用双指针法。
- 初始化一个空的二维向量 ret,用于存储结果。
- 对数组中的每个元素 nums[i] 进行遍历:
- 使用长整型变量 a 存储 target - nums[i],将四数之和问题转化为三数之和问题。
- 对于每个元素 nums[j],其中 j 从 i+1 开始遍历:
- 初始化两个指针 left 和 right,分别指向 j+1 和数组末尾。
- 使用长整型变量 b 存储 a - nums[j]。
- 在 left < right 的条件下,进行循环:
- 计算当前指针所指元素的和 sum = nums[left] + nums[right]。
- 如果 sum < b,则将 left 右移一位。
- 如果 sum > b,则将 right 左移一位。
- 如果 sum == b,说明找到了一组解,将 {nums[left], nums[right], nums[j], nums[i]} 加入到 ret 中。
- 然后将 left 右移一位,right 左移一位,继续寻找下一组解。
- 同时需要进行去重处理,即跳过所有与当前值相同的元素,以防止重复的解。
- 在每次循环结束后,将指针 j 右移一位,继续寻找下一个元素的解。
- 同样需要进行去重处理,跳过所有与当前值相同的元素。
- 在外层循环结束后,将指针 i 右移一位,继续寻找下一个元素的解。
- 同样需要进行去重处理,跳过所有与当前值相同的元素。
- 最后返回结果 ret。
3. 编写代码
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
sort(nums.begin(),nums.end());
vector<vector<int>> ret;
int left,right,sum;
int n=nums.size();
for(int i=0;i<n;)
{
long long a=target-nums[i];
for(int j=i+1;j<n;)
{
left=j+1,right=n-1;
long long b=a-nums[j];
while(left<right)
{
sum=nums[left]+nums[right];
if(sum<b) left++;
else if(sum>b) right--;
else if(sum==b)
{
ret.push_back({nums[left],nums[right],nums[j],nums[i]});
left++;
right--;
while(left<right && nums[left]==nums[left-1])
left++;
while(left<right && nums[right]==nums[right+1])
right--;
}
}
j++;
while(j<n && nums[j]==nums[j-1])
j++;
}
i++;
while(i<n && nums[i]==nums[i-1])
i++;
}
return ret;
}
};