给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
先对特殊情况进行处理,
if(nums==null||nums.length<3){
return Collections.EMPTY_LIST;
}
首先先把该数组进行排序,方便后面操作
然后创建一个将要返回的数组list
然后做一个循环处理,循环的次数为nums.length-2,因为除了i之外还至少需要预留两个数字的位置
循环中首先需要判断如果nums[i]如果大于0了,那么直接可以结束该循环了,因为三数相加必定大于0,
如果i上的元素和上一次循环的i上的元素相同,那么就可能产生同样的集合,为了去重也是需要跳过此次循环
之后定义三个变量,分别为循环i位置的a,i+1位置的b和最后一个位置c.
如果三个数相加等于0,那么把这三个数存到一个新的集合中,然后把这个集合添加到一开始定义的集合list中,之后进行最重要的地方去重,如果b后面,c前面得数和现在一样,那么就进行b++或者c–;
while(b<c&&nums[b]==nums[b+1]){
b++;
}
while(b<c&&nums[c]==nums[c-1]){
c--;
}
这时候便可以b++,c–继续判断是否有加起来等于0的;
如果a+b+c位置上的元素加起来大于0,就c–把右边大的数减小
如果a+b+c位置上的元素加起来小于0,就b++把左边大的数增大
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
if(nums==null||nums.length<3){
return Collections.EMPTY_LIST;
}
Arrays.sort(nums);
List<List<Integer>> list = new ArrayList();
for(int i=0;i<nums.length-2;i++)
if(nums[i]>0){
break;
}
if(i>0&&nums[i]==nums[i-1]){
continue;
}
int a = nums[i];
int b = i+1;
int c = nums.length-1;
while(b<c){
if((a+nums[b]+nums[c])==0){
List temp = new ArrayList();
temp.add(a);
temp.add(nums[b]);
temp.add(nums[c]);
list.add(temp);
while(b<c&&nums[b]==nums[b+1]){
b++;
}
while(b<c&&nums[c]==nums[c-1]){
c--;
}
b++;
c--;
} else if((a+nums[b]+nums[c])<0){
b++;
} else{
c--;
}
}
}
return list;
}
}