15. 3Sum

问题

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

分析：

排序后处理，遍历作为第一个数，另外两个数满足-nums[i] = nums[j] + nums[k]，jk分别是左右游标，如果-nums[i] > nums[j] + nums[k]，则左游标右移动，反之右游标左移。

非常需要注意的是返回的结果中不能有重复，尝试使用HashSet之后转换是超时的。参考Discuss中游标跳过相同值得方式过的。

java

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        //Set<List<Integer>> ansL = new HashSet<>();  
        List<List<Integer>> ansL = new ArrayList<>();
        Arrays.sort(nums);
        for(int i = 0; i < nums.length; i++){
            if(i==0 || i>0 && nums[i] != nums[i-1]){ //去重
                int j = i + 1, k = nums.length - 1;
                while(j < k ){
                    if(nums[i] + nums[j] + nums[k] == 0){
                        ansL.add(Arrays.asList(nums[i], nums[j], nums[k]));
                        while(j < k && nums[j] == nums[j+1] ) j++; //去重
                        while(j < k && nums[k] == nums[k-1] ) k--; //去重
                        j++; k--;
                    }
                    else if(-nums[i] > nums[j] + nums[k]) j++;
                    else k --;
                }
            }
        }
        //List<List<Integer>> ans = new ArrayList<>(ansL);
        return ansL;
    }
}