给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
思路
1、首先对数组进行排序
2、排序后固定一个数 nums[i],再使用左右指针指向nums[i]后面的两端,数字分别为nums[L] 和 nums[R],计算三个数的和 sum 判断是否满足为 0,满足则添加进结果集
3、如果nums[i]大于 0,则三数之和必然无法等于 0,结束循环
4、如果 nums[i]= nums[i-1],则说明该数字重复,会导致结果重复,所以应该跳过
5、当 sum = 0 时,nums[L]= nums[L+1] 则会导致结果重复,应该跳过,L++
6、当 sum =0 时,nums[R]= nums[R-1] 则会导致结果重复,应该跳过,R--
时间复杂度:O(n^2),n 为数组长度
public List<List<Integer>> threeSum(int[] nums) {
// 排序
Arrays.sort(nums);
List<List<Integer>> ans = new ArrayList<>();
if (nums == null || nums.length < 3){
return ans;
}
// 三个数abc肯定要有负数有正数
// 以最左边的数为基准,对右边两个做判断
for (int i = 0; i < nums.length; i++) {
if (nums[i]>0){
break;
}
// 去重
if (i>0 && nums[i]==nums[i-1]){
continue;
}
int L = i+1;
int R = nums.length-1;
while (L<R){
int sum = nums[i]+nums[L]+nums[R];
if (sum > 0){
R--;
}else if (sum < 0){
L++;
}else{
ans.add(Arrays.asList(nums[i],nums[L],nums[R]));
// 对结果去重
while (L<R && nums[L]==nums[L+1]){
L++;
}
while (L<R && nums[R]==nums[R-1]){
R--;
}
// 尝试下一组数字,所以两个都要变
L++;
R--;
}
}
}
return ans;
}