给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 请你找出所有满足条件且不重复的三元组。
注意: 答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
这个题难在去重,要想清楚怎么去重,a,b,c都要去重。a要与a的前一个循环数比较,也就是i-1。如果i-1时的a之前没有出现,可以继续,出现过,则表示这个a已经用过了。
对于b和c要在sum等于0时再判断去重。同理b要和它后面一个数比较,c要和它前面那个数比较。
package Hash;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class LeetCode15_01 {
public static void main(String[] args) {
int[] nums = {-1,0,1,2,-1,4};
List<List<Integer>> result = threeSum(nums);
System.out.println(result);
}
public static List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result = new ArrayList<>();
//先对数组排序
Arrays.sort(nums);
//找出a+b+c=0
//a=nums[i],b=nums[i+1],c=nums[right]
for (int i = 0; i < nums.length; i++) {
//排序后,如果第一个元素大于0,那么不可能符合题意了,返回
if(nums[0]>0){
return result;
}
//对a去重,判断nums[i]==nums[i-1]&&i>0;
if(i>0 && nums[i]==nums[i-1]){
continue;
}
int left = i+1;
int right = nums.length-1;
while (right>left){
int sum = nums[i]+nums[left]+nums[right];
//判断a+b+c与0的大小来调整left和right的位置
if (sum>0){
right--;
}else if (sum<0){
left++;
}else {
result.add(Arrays.asList(nums[i],nums[left],nums[right]));
//对b去重nums[left]==nums[left+1]&&right>left;
while (right>left && nums[left]==nums[left+1]){left++;}
//对c去重nums[right]==nums[right-1]&&right>left;
while (right>left && nums[right]==nums[right-1]){right--;}
right--;
left++;
}
}
}
return result;
}
}