原题链接在这里:https://leetcode.com/problems/3sum/
Two Sum 是这道题的子问题,但是对于3Sum来说又不方便用HashMap,因为不方便存入相同的key,e.g -1,-1,2.
因此,这道题需要先进行排序,对于0 到 length-3 的每一个数后面生成两个指针,一个指向当前元素的下一元素,一个指向最后一个元素,判断三个数的和是否是target,若比target小,头一个指针后移,若比target大,后一个指针前移。
Note: 1. for loop 是注意时候会有溢出,因为会用到至少后面两个数,所以终止条件应该是 i<nums.length - 2;
2. 遇到相同元素时,直接跳过,否者会把相同的 list 存两遍,注意这里是检查是否与上一个元素相同,所以应该比较 nums[i] == nums[i-1],在比较这个之前要加上 i>0,避免overflow.
3. 最后一个else 中记得加入 j++;k--; 否则就是infinite loop, 并且注意j++;k--时是否会遇到相同元素,遇到需跳过。否则会加 入重复的结果,e.g. -2, 0, 0, 2 会加两个 -2,0,2 的 subList.
AC Code:
public class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(nums == null || nums.length < 3)
return res;
Arrays.sort(nums);
for(int i = 0 ; i < nums.length-2; i++){ //error
if(i>0 && nums[i] == nums[i-1]){ //error
continue;
}
int j = i+1;
int k = nums.length - 1;
while(j<k){
if(nums[i] + nums[j] + nums[k] < 0){
j++;
}else if(nums[i] + nums[j] + nums[k] > 0){
k--;
}else{
List<Integer> subList = new ArrayList<Integer>();
subList.add(nums[i]);
subList.add(nums[j]);
subList.add(nums[k]);
res.add(subList);
j++;
k--;
while(j<k && nums[j] == nums[j-1]){ //error
j++;
}
while(j<k && nums[k] == nums[k+1]){
k--;
}
}
}
}
return res;
}
}