给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list = new ArrayList();
int len = nums.length;
if(nums==null||len<3){
return list;
}
Arrays.sort(nums);
for(int i=0;i<len;i++){
if(nums[i]>0){
break;
}
if(i>0&&nums[i]==nums[i-1]){
continue;
}
int left = i+1;
int right = len-1;
while(left<right){
int sum = nums[i]+nums[left]+nums[right];
if(sum==0){
list.add(Arrays.asList(nums[i],nums[left],nums[right]));
while(left<right&&nums[left]==nums[left+1]){
left++;
}
while(left<right&&nums[right]==nums[right-1]){
right--;
}
left++;
right--;
}else if(sum<0){
left++;
}else if(sum>0){
right--;
}
}
}
return list;
}
}
这个解法中,需要注意几个要点; 1.对数组进行排序 Arrays.sort(nums); 2.边界值判断,当nums为null或者长度小于3的判断
if (nums == null || nums.length < 3) {
return ans;
}
3.nums[I]>0时需要跳出循环
if(nums[i] > 0) break;
4.left和right的初始条件定义
int L = i + 1;
int R = len - 1;
5.因为不允许出现重复组合,所以去重,无论是 i还是 left和right都要去重
if(i > 0 && nums[i] == nums[i-1]) continue; // 去重
while (L < R && nums[L] == nums[L + 1]) L++;
while (L < R && nums[R] == nums[R - 1]) R--;