本文记录的是刷题过程中的重要概念和笔记。如有侵权,请联系删除。
第18题. 四数之和
力扣题目链接(opens new window) https://leetcode.cn/problems/4sum/
题意:给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。
注意:
答案中不可以包含重复的四元组。
示例: 给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。 满足要求的四元组集合为: [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]
思路
四数之和,和15.三数之和 (opens new window)是一个思路,都是使用双指针法, 基本解法就是在15.三数之和 (opens new window)的基础上再套一层for循环。
有一些细节需要注意,例如: 不要判断nums[k] > target 就返回了。比如:数组是[-4, -3, -2, -1],target是-10,不能因为-4 > -10而跳过。但是我们依旧可以去做剪枝,逻辑变成nums[i] > target && (nums[i] >=0 || target >= 0)就可以了。
四三数之和的时间复杂度是O(n2),四数之和的时间复杂度是O(n3) 。
那么一样的道理,五数之和、六数之和等等都采用这种解法。
对于15.三数之和 (opens new window)双指针法就是将原本暴力O(n3)的解法,降为O(n2)的解法,四数之和的双指针解法就是将原本暴力O(n4 )的解法,降为O(n3)的解法。
454.四数相加II (opens new window),相对于本题简单很多,因为本题是要求在一个集合中找出四个数相加等于target,同时四元组不能重复。
454.四数相加II (opens new window)是四个独立的数组,只要找到A[i] + B[j] + C[k] + D[l] = 0就可以,不用考虑有重复的四个元素相加等于0的情况,所以相对于本题还是简单了不少!
C++
class Solution
{
public:
vector<vector<int>> fourSum(vector<int> &nums, int target)
{
vector<vector<int>> result;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i++)
{
// 1级剪枝
if (nums[i] > target && nums[i] >= 0)
break;
// if (i > 0 && nums[i] == nums[i - 1]) {
// continue;
// }
int newTarget = target - nums[i];
// 变成三数之和
for (int j = i + 1; j < nums.size(); j++)
{
// 2级剪枝处理
if (nums[j] + nums[i] > newTarget && nums[j] >= 0) {
break;
}
int left = j + 1;
int right = nums.size() - 1;
while (left < right)
{ //
if (nums[j] + nums[left] > newTarget - nums[right])
right--; //
else if (nums[j] + nums[left] < newTarget - nums[right])
left++; //
else
{
result.push_back(vector<int>{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++;
}
}
while (j + 1 < nums.size() && nums[j] == nums[j + 1])
j++; //
}
while (i + 1 < nums.size() && nums[i] == nums[i + 1])
i++;
}
return result;
}
};
- (nums[j] + nums[left] > newTarget - nums[right]) —防止加后越界,分立两边
- while (j + 1 < nums.size() && nums[j] == nums[j + 1]) j++;
使用之后再去重的时候要用while不可用if