四数之和
- 1. 题目解析
- 2. 讲解算法原理
- 3. 编写代码
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;}
};