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;
    }
}