题目描述
给你一个整数数组
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
题目思路
排序+双指针法
先对数组进行排序,排序后的数组,可以直接过滤一些不满足条件的情况,比如数组所有元素都大于0,直接就可以排除
然后固定i,寻找j和k,j从i后边第一个元素开始找,k从末尾元素开始找,由于数组是从小到大排序后的,
所以当nums[i] + nums[j] + nums[k]>0,nums[k],已经是最大了,此时只能是k向左移,和才能变小
当nums[i] + nums[j] + nums[k]<0,此时只能是j向右移,和才能变大
当nums[i] + nums[j] + nums[k]=0,记录下当前值就可以
但是题目要求不能包含重复的三元组
,所以要去掉重复的元素
当nums[i-1]与nums[i]一样时,因为已经将 nums[k - 1]
的所有组合加入到结果中,本次双指针搜索只会得到重复组合。,所以直接结束本次循环,继续下一次即可
同理可以去除重复的j和k
代码及解析
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
//排序
Arrays.sort(nums);
for(int i=0;i<nums.length;i++){
//排序后第一个元素都大于0,直接退出
if(nums[i]>0){
break;
}
//去重i
if(i>0&&nums[i]==nums[i-1]){
continue;
}
//定义左右指针
int l = i+1;
int r = nums.length-1;
while(l<r){
int s = nums[i] + nums[l] + nums[r];
//和大于0,r 左移
if(s>0){
r--;
//和小于0,l右移
}else if(s<0){
l++;
}else{
//去重l
while(l<r&&nums[l]==nums[l+1]){
l++;
}
//去重r
while(l<r&&nums[r]==nums[r-1]){
r--;
}
//结果存入列表
List<Integer> list = new ArrayList<>();
list.add(nums[i]);
list.add(nums[l]);
list.add(nums[r]);
res.add(list);
//继续移动
l++;
r--;
}
}
}
return res;
}
}