LeetCode 137 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:
Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)

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)

分析：

Ksum的问题，因为规定了元素数量，不好用回溯做，因此要列举所有情况，过程中，保证3个元素下标的有序就可以了。

需要注意的是，因为S中可能存在重复的数字，所以即便保证了三个下标的有序，还是有可能找到重复的结果，所以需要一个HashSet来判断重复。

public class Solution {
    public List<List<Integer>> threeSum(int[] num) {
        List<List<Integer>> res  = new ArrayList<List<Integer>>();
        if(num == null || num.length < 3) return res;
        HashSet<List<Integer>> set = new HashSet<List<Integer>>();
        Arrays.sort(num);
        
        for(int i=0; i< num.length-2; i++){
            int j=i+1;
            int k=num.length-1;//利用排序的特点减小复杂度
            while(j<k){
                int sum = num[i]+num[j]+num[k];
                if(sum<0)
                    j++;
                else if(sum>0)
                    k--;
                else{
                    List<Integer> item = new ArrayList<Integer>();
                    item.add(num[i]);
                    item.add(num[j]);
                    item.add(num[k]);
                    //因为num可能存在重复元素，所以i,j,k的组合也可能重复
                    if(!set.contains(item)){
                        res.add(item);
                        set.add(item);
                    }
                    
                    j++;
                    k--;
                }
            }
        }
        return res;
    }
}