[leetcode]3Sum

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)

思路1：
与之前的[leetcode]Two Sum的思路1相似，加一层外循环。时间复杂度为O(n^3)
思路2：
与之前的[leetcode]Two Sum的思路1相似，加一层外循环。时间复杂度为O(n^2)，思路完全一样，大家可以自己实现一下。
思路3：
先对整个数组进行排序，外层循环对每个元素i(i < length - 2)进行遍历，然后在余下的元素(i + 1 ~ length)中采用双指针进行检索，参考原博文[leetcode]3Sum 的讲解
代码如下：

public class Solution {
    public List<List<Integer>> threeSum(int[] num) {
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        if(num == null || num.length < 3) return result;
        Arrays.sort(num);
        for(int i = 0; i < num.length; i++){
            if(i > 0 && num[i] == num[i - 1]) continue;
            int begin = i + 1;
            int end = num.length - 1;
            while(begin < end){
                if(num[begin] + num[end] + num[i] == 0){
                    List<Integer> list = new ArrayList<Integer>();
                    list.add(num[i]);
                    list.add(num[begin]);
                    list.add(num[end]);
                    result.add(list);
                    begin++;
                    end--;
                    while(begin < end && num[begin] == num[begin - 1]) begin++;
                    while(end > i && num[end] == num[end + 1]) end--;
                }else if(num[begin] + num[end] + num[i] > 0){
                    end--;
                }else{
                    begin++;
                }
            }
        }
        return result;
    }
}