四数之和
- 题目
- 思路
- 代码
题目
思路
类似于三数之和,先排序,利用双指针解题。
如果排序后的第一个元素大于目标值,直接返回,为什么nums[i]需要大于等于0,因为目标值可能为负数。比如:数组是[-4, -3, -2, -1],target是-10,不能因为-4 > -10而跳过。
四数之和的双指针解法是两层for循环nums[i] + nums[j]为确定值,依然是循环内有left和right下标作为双指针,找出nums[i] + nums[j] + nums[left] + nums[right] == target的情况
去重和三数之和一样很简单,i和j如果重复了怎么办,它们是nums里遍历的元素,那么应该直接跳过去。nums[i+ 1]是j,nums[j+1]是left,这些都是要去重的,所以我们考虑nums[i-1]和nums[j-1]。
left和right去重就很简单了,因为我们已经确定了前两个值,要想确定最终的目标值,如果left确定了,那么right就是固定的,所以left指针和left+1相等时,我们直接进行left++就可以了,同理,right类似。
代码
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
//返回二维数组
vector<vector<int>> res;
//排序,方便后续使用双指针
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i++) {
//如果排序后第一个元素大于目标值,直接返回,为什么需要大于等于0,因为目标值可能为负数
if (nums[i] > target && nums[i] >= 0) {break;}
//对i去重
if (i > 0 && nums[i] == nums[i - 1]) continue;
for (int j = i + 1; j < nums.size(); j++) {
if (nums[i] + nums[j] > target && nums[i] >= 0) {break;}
//对j去重 注意 j是从i+1开始的
if (j > i + 1 && nums[j] == nums[j - 1]) continue;
//开始双指针循环查找
int left = j + 1;
int right = nums.size() - 1;
while (right > left) {
if ((long)nums[i] + nums[j] + nums[left] + nums[right] > target) {
right--;
} else if ((long)nums[i] + nums[j] + nums[left] + nums[right] < target) {
left++;
} else {
res.push_back(vector<int>{nums[i], nums[j], nums[left], nums[right]});
//对left和right去重
while (right > left && nums[left] == nums[left + 1]) left++;
while (right > left && nums[right] == nums[right - 1]) right--;
//一轮循环完,双指针向内收缩
right--;
left++;
}
}
}
}
return res;
}
};