import java.util.*;
class Solution {
public List<List<Integer>> threeSum(int[] sum) {
List<List<Integer>> result=new LinkedList<>();
if (sum==null||sum.length<3) {
return result;
}
Arrays.sort(sum);//sort
int len=sum.length;
for(int i=0;i<len-2;i++){
if(i>0&&sum[i]==sum[i-1]){
continue;
}
int left=i+1;
int right=len-1;
while(left<right){
int res=sum[i]+sum[left]+sum[right];
if (res>0) {
right--;
}else if (res<0) {
left++;
}else{
result.add(Arrays.asList(sum[i],sum[left],sum[right]));
while(left<right&&sum[left]==sum[left+1]){
left++;
}
while(left<right&&sum[right]==sum[right-1]){
right--;
}
left++;
right--;
}
}
}
return result;
}
}
属于two sum问题的升级版本。首先最暴力的方法是使用三层for循环,但是对应的时间复杂度为O(N^3),不是一个很好的方法。所以,需要考虑一下使用额外的辅助空间来解决这个问题。一种方式是先从第一个值开始,遍历后续数组,将二者相加,对应的再去数组中找使得三者之和为0的值。需要注意去重操作,当时我一直不通过的原因是最中的集合之中存在重复的列表,-1,0,-1出现了两次。因为数组中-1出现了两次,使用if(i>0&&sum[i]==sum[i-1]){continue;}来进行去重操作。很关键,当然,如果这里使用一个set来进行去重也是可以的。
附一个使用set的代码。
public List<List<Integer>> threeSum(int[] nums) {
Set<List<Integer>> res = new HashSet<>();
if(nums.length==0) return new ArrayList<>(res);
Arrays.sort(nums);
for(int i=0; i<nums.length-2;i++){
int j =i+1;
int k = nums.length-1;
while(j<k){
int sum = nums[i]+nums[j]+nums[k];
if(sum==0)res.add(Arrays.asList(nums[i],nums[j++],nums[k--]));
else if ( sum >0) k--;
else if (sum<0) j++;
}
}
return new ArrayList<>(res);
}