leetcode解题方案--015--3 sum

题目

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

分析

题目隐含的条件，不能有重复的list，不能每个小list必须从小到大排序。

我的思路有两个
1 将n^3转化为2sum。leetcode第一题就是一道2sum，我用hashmap做的
2 使用数组左右指针。

实践来看，第二种方法更优。

这个有两个测试例没过。
稍后我会分析一下时间长的原因。

class Solution {     public List<List<Integer>> threeSum(int[] nums) {        HashSet<List<Integer>> set = new HashSet<>();        List<List<Integer>> list = new ArrayList<>();        for (int i = 0; i < nums.length; i++) {            Map<Integer, Integer> map = new HashMap<Integer, Integer>();            for (int k = i + 1; k < nums.length; k++) {                if (map.containsKey(nums[k])) {                    List<Integer> tmpList = new ArrayList<>();                    tmpList.add(nums[i]);                    tmpList.add(nums[k]);                    tmpList.add( 0 - nums[i] - nums[k]);                    Collections.sort(tmpList);                    set.add(tmpList);                } else {                    map.put(0 - nums[i] - nums[k], k);                }            }        }        list.addAll(set);        return list;    }}

这是左右指针的方法，去掉了放进set去重的部分，时间更短。
那如何保证不重复呢，关键点在于排序。一个有序的数组，相同的数就会相邻。
在这个程序中 if (i == 0 || (i > 0 && nums[i] != nums[i-1]))会去掉-sum相同的情况，而while (k1 < k2 && nums[k1] == nums[k1+1]) k1++;
while (k1 < k2 && nums[k2] == nums[k2-1]) k2–;会去掉指针移动时相同的情况。

 public static List<List<Integer>> threeSum(int[] nums) {        Arrays.sort(nums);        List<List<Integer>> list = new ArrayList<>();        for (int i = 0; i < nums.length - 2; i++) {            if (i == 0 ||  (i > 0 && nums[i] != nums[i-1])) {                int k1 = i + 1;                int k2 = nums.length - 1;                while (k2-k1>=1) {                    if (nums[i] + nums[k1] + nums[k2] == 0) {                        list.add(Arrays.asList(nums[i], nums[k1], nums[k2]));                        while (k1 < k2 && nums[k1] == nums[k1+1]) k1++;                        while (k1 < k2 && nums[k2] == nums[k2-1]) k2--;                        k1++;k2--;                    } else if (nums[i] + nums[k1] + nums[k2] > 0) {                        k2--;                    }else {                        k1++;                    }                }            }        }        return list;    }