Three sum

Posted 2020-10-14

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]
]

1.暴力解法
时间复杂度高达O(n^3)

public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> lists =new ArrayList<List<Integer>>();
        for(int i=0;i<nums.length;i++) {
            for (int j = i + 1; j < nums.length; j++) {
                for(int z=j+1;z<nums.length;z++){
                    if(nums[i]+nums[j]+nums[z]==0){
                        List<Integer> list =new ArrayList<Integer>();
                        list.add(nums[i]);
                        list.add(nums[j]);
                        list.add(nums[z]);
                        lists.add(list);
                    }
                }
            }
        }
        return lists;
    }

2.使用map

时间复杂度为O(n+n^2)，即O(n^2)

    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> lists =new ArrayList<List<Integer>>(); 
        Map<Integer,Integer> map =new HashMap<Integer,Integer>();
        for(int i=0;i<nums.length;i++){
            map.put(nums[i],i);
        }

        for(int i=0;i<nums.length;i++){
            for(int j=i+1;j<nums.length;j++) {
                int res=0-nums[i]-nums[j];
                if(map.containsKey(res)&&map.get(res)!=i&&map.get(res)!=j){
                    List<Integer> list =new ArrayList<Integer>();
                    list.add(res);
                    list.add(nums[i]);
                    list.add(nums[j]);
                    lists.add(list);
                }
            }
        }
        return lists;
    }

这个的运行结果让人头疼。

有没有好的办法可以排重，如果这样的话：

    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> lists =new ArrayList<List<Integer>>();
        Map<Integer,Integer> map =new HashMap<Integer,Integer>();

        for(int i=0;i<nums.length;i++){
            for(int j=i+1;j<nums.length;j++) {
                int res=0-nums[i]-nums[j];
                if(map.containsKey(res)&&map.get(res)!=i&&map.get(res)!=j){
                    List<Integer> list =new ArrayList<Integer>();
                    list.add(res);
                    list.add(nums[i]);
                    list.add(nums[j]);
                    lists.add(list);
                }
                map.put(nums[i],i);
            }
        }
        return lists;
    }

运行结果：

重写equals方法和hashcode方法对结果进行过滤，可不可以？夜深了，就到这里。明天解决。