八、三数之和
题目:
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
注意: 答案中不可以包含重复的三元组。
示例1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
提示:
- 3 <= nums.length <= 3000
- -105 <= nums[i] <= 105
解题思路分析:
暴力解法:
三层for循环遍历数组来找三元组。
哈希法:
两层for循环就可以确定 a 和b 的数值了,可以使用哈希法来确定 0-(a+b) 是否在 数组里出现过,其实这个思路是正确的,但是我们有一个非常棘手的问题,就是题目中说的不可以包含重复的三元组。
注意:三元组是去重的!
使用哈希法的复杂之处在于去重;由于没有顺序,去重有些困难。
(自己做遇到的问题:由于元素的出现顺序不同,直接用set并不能实现去重;)
双指针法:
其实这道题目使用哈希法并不十分合适, 因为在去重的操作中有很多细节需要注意,在面试中很难直接写出没有bug的代码。
而且使用哈希法 在使用两层for循环的时候,能做的剪枝操作很有限,虽然时间复杂度是O(n^2),也是可以在leetcode上通过,但是程序的执行时间依然比较长 。
这道题目使用双指针法 要比哈希法高效一些:
使用双指针法一定要先对数组进行排序;
得到有序数组(从小到大排序) 后:
- 一个for循环去遍历i的位置(从下标0开始),得到a;
- 一个指针left,指向i的后一个位置,得到b;
- 一个指针right,指向数组最后一个位置,得到c;
- 如果a+b+c>0,说明三数之和大了,就让right往左移动一位;a+b+c<0,说明三数之和小了,就让left往右移动一位;
- 如果找到a+b+c=0,就把他们加进result数组中(二维数组)移动i到下一个位置;
- 如果找不到a+b+c=0,就直到left遇到right后,移动i到下一个位置;
- 继续循环如上操作。
细节在于去重:
- a、b、c三个数都要做去重操作;例如不同位置得到的-1、-1、2,算做一个三元组,而不是多个
- 具体操作在代码中讲解。
代码讲解:
//定义一个List,用来存放找到的三元组
List<List<Integer>> result = new ArrayList<>();
//对数组进行排序
Arrays.sort(nums);
//循环遍历数组
for(int i = 0; i < nums.length; i++) {
//如果排序后的第一个数都大于0,代表数组中数都大于0,相加不可能为0了
if(nums[i] > 0) return;
//对a进行去重
if(i > 0 && nums[i] == nums[i - 1]) {
continue; //跳过当前这一次循环
}
int left = i + 1;
int right = nums.length - 1;
//移动left和right指针
while(left < right) {
if(nums[i] + nums[left] + nums[right] > 0) right--;
else if(nums[i] + nums[left] + nums[right] < 0) left++;
else {
result.put(Arrays.asList(nums[i],nums[left],nums[right]);
//对b、c进行去重,去重操作应该放到找到一个三元组之后
while(right > left && nums[right] == nums[right - 1 ]) right--;
while(right > left && nums[left] == nums[left + 1 ]) left++;
//收获了一个三元组以后,left和right应该同时往中间移动一个
right--;
left++;
}
}
}
return result;
去重的逻辑:
要求的是三元组不能重复,但是三元组里面的元素是可以重复的(如:0,0,0)
a 如果重复了怎么办,a是nums里遍历的元素,那么应该直接跳过去。
给a去重:
是应该判断nums[i] == nums[i+1],还是应该判断nums[i] == nums[i-1] ?
如果判断的是nums[i] == nums[i+1],则相当于判断i 和left的值是否相等,这相当于在判断i和结果集中的其他元素是否相等,如果这样判断的话,就去掉了结果集里面的重复元素,是不正确的;
例如:-1,-1,2是一个结果集;
那么i指向-1,left指向-1,right指向2;如果i和i+1(也就是left)作对比了,那么就会直接跳过这个结果集;
应该判断的是 nums[i] == nums[i-1] ;
如果i的前一位和目前i的值一样,则直接跳过这个,将i指向下一个位置;
例子:如果数组中是-2,-1,-1,-1,2,3….
当i指向第一个-1的时候,只要前一位不是-1,就不跳过,那么 {-1, -1 ,2} 这组数据一样可以收录到结果集里;
当i指向第二个-1的时候,它和前一位-1相等,就会直接跳过,将i指向第三个-1;
当i指向第三个-1的时候,它和前一位-1相等,就会直接跳过,将i指向2;
当i指向2的时候,它和前一位-1不相等了,就不跳过, 开始找三元组;
给b和c去重:
b和c的去重操作一定要放在找到三元组之后,要不然会直接把结果集也跳过了;
找到一个三元组后,如果下一个right和目前找到的三元组中的right相等,right指向下一个元素(相等的元素);
与此同时,如果left的下一个和目前的left相等,则直接指向下一个(相等的元素);
相当于去重操作结束以后,left和right都指向了最后一个和他们目前的值相等的地方。
举个例子:
指针如图所示的时候,找到了一个三元组-1、-1、2;
那么进行完b和c的去重操作后,指针如下图所示:
所以找到一个三元组,进行去重操作以后,还需要把left和right都向中间移动一个位置,继续找下一个三元组。
总结:
重难点在于去重的操作,要理解去重的逻辑。
题解:
双指针解法: