给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
第一次的解决方案,可以得到结果但是没有通过时间限制。
public static List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list=new ArrayList<>();
for(int i=0;i<nums.length;i++) {
for(int j=i+1;j<nums.length;j++) {
int tmp=-nums[i]-nums[j];
for(int k=j+1;k<nums.length;k++) {
if(nums[k]==tmp) {
List<Integer> li=new ArrayList<>();
li.add(nums[i]);
li.add(nums[j]);
li.add(nums[k]);
list.add(li);
}
}
}
}
for(int i=0;i<list.size();i++) {
for(int j=i+1;j<list.size();j++) {
if(list.get(i).containsAll(list.get(j))&&list.get(j).containsAll(list.get(i))) {
list.remove(j--);
}
}
}
return list;
}
第二次的解决法案,通过时间测试(思路不赘述,看代码)
public static List<List<Integer>> threeSum2(int[] nums) {
List<List<Integer>> list=new ArrayList<>();
Arrays.sort(nums);
for(int k=0;k<nums.length;k++) {
if(nums[k]>0) {
break;
}
if(k>0&&nums[k]==nums[k-1]) {
continue;
}
int target=0-nums[k];
int i=k+1,j=nums.length-1;
while(i<j) {
if(nums[i]+nums[j]==target) {
List<Integer> li=new ArrayList<>();
li.add(nums[i]);
li.add(nums[j]);
li.add(nums[k]);
list.add(li);
while(i<j&&nums[i]==nums[i+1]) {
i++;
}
while(i<j&&nums[j]==nums[j-1]) {
j--;
}
i++;
j--;
}else if(nums[i]+nums[j]<target) {
i++;
}else {
j--;
}
}
}
return list;
}