三数之和
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[ [-1, 0, 1],[-1, -1, 2]]
原来解法(通过了311/313个测试用例,超出执行时间):
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
Set<List<Integer>> result = new LinkedHashSet<>();
int i,j,k;
for(i=0;i<nums.length-2;i++){
if(nums[i]>0) break;
for(j=i+1;j<nums.length-1;j++){
for(k=j+1;k<nums.length;k++){
if(nums[i]+nums[j]+nums[k]==0){
List<Integer> value = Arrays.asList(nums[i], nums[j], nums[k]);
result.add(value);
}
}
}
}
return new ArrayList<>(result);
}
}
改进后的解法(参考题解):
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> result=new ArrayList<>();
for(int i=0;i<nums.length-2;i++){
if(nums[i]>0) break;
if(i>0&&nums[i]==nums[i-1]) continue;
int j=i+1,k=nums.length-1;
while(j<k){
int sum=nums[i]+nums[j]+nums[k];
if(sum==0){
result.add(new ArrayList<Integer>(Arrays.asList(nums[i],nums[j],nums[k])));
while(j<k&&nums[j]==nums[++j]);
while(j<k&&nums[k]==nums[--k]);
}
else if(sum>0){
while(j<k&&nums[k]==nums[--k]);
}
else{
while(j<k&&nums[j]==nums[++j]);
}
}
}
return result;
}
}
思路:先通过sort对数组从小到大排序,nums[i]如果是正数就终止,然后用双指针j和k,从两端向中间运行,如果三数之和小于0,左指针j向右移动一位;三数之和大于0,右指针k向左移动一位;等于0的话则同时移动一位。