之所以放在一起是因为,"四数之和"的解题方法基本与"三数之和"一致

由此我们可以推出n数之和的解法

本质上,我们只是使用双指针的方法降低此类问题的时间复杂度

当然用哈希法也可以解,那就是另外的故事了

三数之和

力扣题目链接(opens new window)

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。

注意: 答案中不可以包含重复的三元组。

示例:

给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]

初见分析

很像四数相加||啊

那题是用了map去解决的,于是照猫画虎开始做

做到最后会发现,题目要求的是返回由数组中三个相加为0的元素构成元组

这里就的有问题了,如果按照原来的思路

那么如何保存符合条件的遍历值的下标?并且还没有重复值

思路

不整哈希法了,用双指针法解会好理解一些,过程如下图所示:

题目要求是返回三个相加为0的数,即: a+b+c = 0,与下标无关

初始化指针

还是分开处理,初始化三个指针

a 对应 i(固定值)

b 对应 left

c 对应 right

这里在使用双指针前需要对数组进行排序(用sort就行不用自己写)

遍历数组

在遍历过程中会出现以下几种情况:

0、nums[i] > 0

​ 这种情况的话就直接return,因为如果第一个数是大于0的数,那么后面的数肯定都是正数(因为提前经过排序),则不可能再计算得到0

1、nums[i] + nums[left] + nums[right] > 0

i 对应的数是最小的,不用管,主要看 bc (所以才叫双指针而不是三指针)

​ 因为已经排好序,所以要让整体值变小进而靠近0,需要让最大的值即right向左移动

2、nums[i] + nums[left] + nums[right] < 0

​ 同理,要让整体值变大,最小的值left要向右移动

3、nums[i] + nums[left] + nums[right] = 0

​ 找到目标三元组,保存

大致结构如下:

class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
...
for (int i = 0; i < nums.size(); i++) { if (nums[i] > 0) {
return result;
}
//去重
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
} int left = i + 1;
int right = nums.size() - 1; while (right > left) {
if (nums[i] + nums[left] + nums[right] > 0) right--;
else if (nums[i] + nums[left] + nums[right] < 0) left++;
else {
//先保存,再去重
}
}
}
}
};
去除重复值

还是以a+b+c为例

去除a中的重复值

其中,i 为最外层循环,每当 i 更新(向后移动)时,需要对其进行去重操作

注意,去重时需要使用num[ i ]和num[ i + 1 ]来比较,而不是使用num[ i - 1 ]和num[ i ]

后者会漏掉某些情况,例如

{-1, -1, 2}
↑ ↑
i-1 i

此时,num[ i - 1 ] = num[ i ],理应直接返回,不再继续遍历

但是这样就会把{-1, -1, 2}这种情况漏掉

去除b、c中的重复值

当第一次满足nums[i] + nums[left] + nums[right] = 0条件时,我们先要对本次的数做保存

之后再满足条件就要对值进行去重操作

原理和对 a 的做法是一样的,仍然是拿当前遍历值与它的下一个值做比较,如果相等就继续移动对应的指针

代码

class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;//初始化一个vec用来存放结果
//先对数组进行排序
sort(nums.begin(),nums.end());
//遍历数组(最外层,i)
for(int i = 0; i < nums.size(); i++){
//判断i是否大于零
//大于零直接返回了
//这种认为规避多余运算的处理称为"剪枝"
if(nums[i] > 0){
return res;
}
//对i进行去重
if(i > 0 && nums[i] == nums[i - 1]){//不要忘了i > 0的条件,因为第一次是不需要去重的
continue;//i往后移
}
//初始化双指针(代表b、c)
int left = i + 1;//指向i的后一位
int right = nums.size() - 1;//指向数组末尾
while(left < right){ //不能取等于号,那样b和c就是同一个值了,不满足三个数的要求
//判断是否找到满足条件的三个值
//三种情况
if(nums[i] + nums[left] + nums[right] > 0){
//右指针左移,减小整体值
right--;
}else if(nums[i] + nums[left] + nums[right] < 0){
//左指针右移,增大整体值
left++;
}else{//等于的情况,找到三个目标值
//先保存第一次满足条件的结果
res.push_back(vector<int>{nums[i], nums[left], nums[right]});
//第二次就开始去重
// //b去重(left)
// if(left < right && nums[left] == nums[left + 1]){
// left++;
// }
// //c去重(right)
// if(left < right && nums[right - 1] == nums[right]){
// right--;
// }
//这里不要用if
//将所有可能的b、c值都做去重判断,直到不满足left < right,结束本次遍历,更新i进行下次遍历
while(left < right && nums[left] == nums[left + 1]){
left++;
}
while(left < right && nums[right - 1] == nums[right]){
right--;
}
//如果没找到重复值,左右指针均向中心移动
left++;
right--;
}
}
}
return res;
}
};
易错点
1、初始化vector

因为这个vector是用来存放一个三个元素的数组的,所以初始化的时候数据类型应该是vector,而不是int

解题模板也有提示,注意看

2、去重的时机

无论是对a还是b、c进行去重,第一次得到的结果都是不用进行去重的

3、保存结果后的去重操作

当第一次满足条件时:

{-1, -1, -1, 3, 2, 2}
↑ ↑ ↑
i left right

我们得到了

这时left、right移动

{-1, -1, -1, 3, 2, 2}
↑ ↑ ↑
i left right

发现nums[left] == nums[left + 1]并且nums[right - 1] == nums[right]

显然是出现了重复,此时需要跳过当前left、right指向的值

直到不出现这种重复情况为止,因此这里必须使用while而不是if

四数之和

力扣题目链接(opens new window)

题意:给定一个包含 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] ]

思路

和三数之和的思路一样,都是用双指针法去解

无非就是再多套一层循环

解决三/四数之和,或者说n数之和问题,本质上是利用双指针的方法去降低时间复杂度

四数之和的双指针解法是两层for循环nums[k] + nums[j]为确定值,依然是循环内有left和right下标作为双指针,找出nums[k] + nums[j] + nums[left] + nums[right] == target的情况,三数之和的时间复杂度是O(n2),四数之和的时间复杂度是O(n3)

简单来说,这里的固定值有两个: k 和 j

代码

class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> res;
//先对数组进行排序
sort(nums.begin(),nums.end()); //剪枝操作
for(int k = 0; k < nums.size(); k++){
//因为target是自定义的,可以dayu0也可以小于0,所以不能单纯的通过nums[k] > target来剪枝
//
if (nums[k] > target && nums[k] >= 0) {
break; // 这里使用break,统一通过最后的return返回
} //对k进行去重
if(k > 0 && nums[k] == nums[k - 1]){//不要忘了k > 0的条件,因为第一次是不需要去重的
continue;//i往后移
} for(int j = k + 1; j < nums.size(); j++){
// 2级剪枝处理,同理
if (nums[k] + nums[j] > target && nums[k] + nums[j] >= 0) {
break;
}
//对j进行去重
if(j > k + 1 && nums[j] == nums[j - 1]){
continue;//j往后移
}
//初始化双指针(代表b、c)
int left = j + 1;//指向j的后一位
int right = nums.size() - 1;//指向数组末尾
while(left < right){
if((long)nums[k] + nums[j] + nums[left] + nums[right] > target){
right--;
}else if((long)nums[k] + nums[j] + nums[left] + nums[right] < target){
left++;
}else{
res.push_back(vector<int>{nums[k], nums[j], nums[left], nums[right]});
//去重
//将所有可能的b、c值都做去重判断,直到不满足left < right,结束本次遍历
while((long) left < right && nums[left] == nums[left + 1]){
left++;
}
while((long) left < right && nums[right - 1] == nums[right]){
right--;
}
//如果没找到重复值,左右指针均向中心移动
left++;
right--;
}
}
}
}
return res;
}
};
注意点
1、剪枝操作

不能使用nums[k] > target直接进行剪枝操作

例如有:[-4, -1, 0, 0],target = -5

此时nums[k] = -4,nums[k]确实大于target

如果按照上面的条件进行剪枝,该结果集会被直接跳过,显然是不行的

因此这种情况下就不要剪枝了,当target大于0时我们再使用nums[k] > target来判断是否剪枝

综上,本题中的剪枝条件应该改为`nums[k] > target && nums[k] >= 0

最新文章

  1. python+uwsgi导致redis无法长链接引起性能下降问题记录
  2. 给Source Insight做个外挂系列之二--将本地代码注入到Source Insight进程
  3. Eclipse中tomcat之后,tomcat的相关配置会被Eclipse重置
  4. MyEclipse设置默认的目光格式
  5. Android Training精要(四) Intent注意事项
  6. mybatis11 sqlMapConfig.xml文件说明
  7. Delphi动态申请数组内存的方法(不使用SetLength,采用和C相似的方式)
  8. 如何识别SQL Server中的IO瓶颈
  9. Webi Report 展示 图片链接 (Image Link)
  10. A low-cost wear-leveling algorithm for block-mappingsolid-state disks
  11. C#--深入理解类型
  12. Mysql表的约束设计和关联关系设计
  13. 编写高质量的Python代码系列(一)之用Pythonic方式来思考
  14. python note 11 函数名的使用、闭包、迭代器
  15. CSRF 攻击(跨域攻击)
  16. Fluentd初探 简介与安装
  17. spark 2.3 导致driver OOM的一个SparkPlanGraphWrapper源码的bug
  18. Heap堆分析(堆转储、堆分析)
  19. [Android] 对自定义图片浏览器经常内存溢出的一些优化
  20. python算两个时间之间的天数,将天数转成int型

热门文章

  1. 用 VS Code 搞 Qt6:让信号和槽自动建立连接
  2. 廖---list tuple dic set
  3. 第一百零八篇:最常用的基本数据类型(Number,String类型)
  4. c#显示和隐藏另外一个进程
  5. .NET周报【11月第2期 2022-11-15】
  6. zabbix6.0安装
  7. halcon如何识别硬币?
  8. Java开发学习(四十三)----MyBatisPlus查询语句之查询投影
  9. 基于.NetCore开发博客项目 StarBlog - (20) 图片显示优化
  10. FCOS论文复现:通用物体检测算法