力扣第18题 四数之和 c++双指针注释版。
题目
中等
给你一个由 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]]
提示:
1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109
思路和解题方法
首先,定义了一个函数
fourSum
,它接受一个整数数组nums
和目标值target
作为参数,并返回一个二维整数数组ans
。在函数内部,创建一个空的二维整数数组
ans
用于存储结果。对输入数组
nums
进行升序排序,以便后续处理中的双指针方法能够正确工作。使用两层嵌套循环遍历数组。外层循环变量
k
表示第一个数的索引,内层循环变量i
表示第二个数的索引。在每次循环开始时,首先进行一些判断和优化操作。如果当前
nums[k]
大于target
并且大于等于0,那么后续的数都会比目标值大,循环可以提前终止。此外,如果k
大于0并且nums[k]
与前一个数相同,说明已经处理过了,可以跳过当前循环。内层循环中,设置两个指针
left
和right
,分别指向剩下的范围中最左边和最右边的数。通过不断调整left
和right
,寻找满足条件的四元组。在指针调整的过程中,先判断四数之和与目标值的大小关系,如果大于目标值,则将右指针向左移动;如果小于目标值,则将左指针向右移动;如果等于目标值,则将四个数添加到结果数组
ans
中,并继续调整指针寻找下一个可能的四元组。在指针调整的同时,还需要注意去重操作。如果当前指针所指的数与前一个数相同,说明已经处理过了,在这种情况下,需要将指针向前移动一位,避免出现重复的四元组。
复杂度
时间复杂度:
O(n^3)
内部循环的复杂度为O(N),外部循环的复杂度也为O(N),而指针调整过程中的while循环的复杂度为O(N)。所以总体的时间复杂度为O(N^3)。
空间复杂度
O(1)
空间复杂度方面,除了存储结果的二维数组
ans
之外,算法并没有使用额外的空间,因此空间复杂度为O(1)。需要注意的是,虽然代码中使用了一个二维结果数组
ans
来存储满足条件的四元组,但这并不会影响算法的空间复杂度。因为最终返回的结果本身需要占用空间来存储,而ans
只是临时存储中间结果的容器,不会随着输入规模的增加而增加额外的空间消耗。
c++ 代码
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> ans; // 存储结果的二维数组
sort(nums.begin(), nums.end()); // 排序数组
for (int k = 0; k < nums.size(); k++) { // 第一个数的循环
if (nums[k] > target && nums[k] >= 0) // 如果第一个数大于目标值且大于等于0,提前终止循环
break;
if (k > 0 && nums[k] == nums[k - 1]) // 如果当前数与前一个数相同,跳过当前循环,避免重复结果
continue;
for (int i = k + 1; i < nums.size(); i++) { // 第二个数的循环
if (nums[k] + nums[i] > target && nums[k] + nums[i] >= 0) // 如果前两个数之和大于目标值且大于等于0,提前终止循环
break;
if (i > k + 1 && nums[i] == nums[i - 1]) // 如果当前数与前一个数相同,跳过当前循环,避免重复结果
continue;
int left = i + 1; // 左指针指向第三个数
int right = nums.size() - 1; // 右指针指向最后一个数
while (right > left) { // 双指针的调整
if ((long)nums[k] + nums[i] + nums[left] + nums[right] > target) { // 四数之和大于目标值,右指针左移
right--;
} else if ((long)nums[k] + nums[i] + nums[left] + nums[right] < target) { // 四数之和小于目标值,左指针右移
left++;
} else { // 四数之和等于目标值
ans.push_back(vector<int>{nums[k], nums[i], nums[left], nums[right]}); // 添加结果到结果数组
// 去重:跳过与当前数相同的数
while (right > left && nums[right] == nums[right - 1]) {
right--;
}
while (right > left && nums[left] == nums[left + 1]) {
left++;
}
right--; // 右指针左移
left++; // 左指针右移
}
}
}
}
return ans; // 返回结果数组
}
};
觉得有用的话可以点点赞,支持一下。
如果愿意的话关注一下。会对你有更多的帮助。
每天都会不定时更新哦 >人< 。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· winform 绘制太阳,地球,月球 运作规律
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)