给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> ans = new ArrayList<List<Integer>>();
//排序
Arrays.sort(nums);
//当数组长度不到3的时候直接返回
if(nums.length < 3){
return ans;
}
for(int i=0;i<nums.length;i++){
int left = i + 1;
int right = nums.length - 1;
//如果数组从第二位开始当前位和当前位前一位数值相同则跳过当前循环
if(i > 0 && nums[i] == nums[i-1]) continue;
while(left < right){
if(nums[i] + nums[left] + nums[right] == 0){
List<Integer> list = new ArrayList<Integer>();
list.add(nums[i]);
list.add(nums[left]);
list.add(nums[right]);
ans.add(list);
right--;
}else if(nums[i] + nums[left] + nums[right] > 0){
//右侧最大下标数值过大需要左移
right--;
}else{
//左侧最小下标数值过小需要右移
left++;
}
}
}
ans = ans.stream().distinct().collect(Collectors.toList());
return ans;
}
}
思路:
特判,对于数组长度 nn,如果数组为 null或者数组长度小于 3,返回 []。
对数组进行排序。
遍历排序后数组:
若 nums[i]>0:因为已经排序好,所以后面不可能有三个数加和等于 0,直接返回结果。
对于重复元素:跳过,避免出现重复解
令左指针 L=i+1 ,右指针 R=n-1,当 L<R 时,执行循环:
当 nums[i]+nums[L]+nums[R]==0,执行循环,判断左界和右界是否和下一位置重复,去除重复解。并同时将 L,RL,R 移到下一位置,寻找新的解
若和大于 0,说明 nums[R] 太大,R 左移
若和小于 0,说明 nums[L] 太小,L右移