排序+双指针
排序,方便区分正负数。三个数和为0必定有一个是小于等于0。
设三个变量,i在小于等于0的区间里遍历,j初始位置是i+1,k初始位置是nums.length-1
i在小于等于0的区间里遍历,每在一个位置,都让j,k遍历数组直到j不小于k.i在遍历时还要注意去重。
当三个数的和小于0,j++
当三个数的和大于0,k–
当三个数的和等于0,保存这三个数,j++、k–(注意nums[k-1] == nums[k]时,要跳过k,避免重复计算)
有两个地方需要去重:
- i在小于等于0的区间里遍历时,当nums[i]与nums[i-1]相同时要跳过i。
- j、k的位置更新时,与i一样注意避免计算相同的nums[j]和nums[k]。
import java.util.ArrayList;
import java.util.Arrays;
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
if(nums.length < 3){
return res;
}
Arrays.sort(nums);
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;
int k = nums.length - 1;
while(j < k){
int sum = nums[i] + nums[j] + nums[k];
if(sum < 0){
j++;
}else if(sum > 0){
k--;
}else{
List<Integer> li = new ArrayList<>();
li.add(nums[i]);
li.add(nums[j]);
li.add(nums[k]);
res.add(li);
while(j < k && nums[k-1] == nums[k]){//去重
k--;
}
while(j < k && nums[j+1] == nums[j]){//去重
j++;
}
k--;//很重要,否则超时,还没明白为什么超时。
j++;
}
}
}
return res;
}
}