[Leetcode] 3Sum

这道题目实际上有非常好的O（N^2)的解法。

关键是利用了2sum的线性解法

public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new LinkedList<List<Integer>>();
        if(nums==null||nums.length==0) return res;
        Arrays.sort(nums);
        for(int i=0;i<nums.length-2;i++){
            if(i!=0&&nums[i-1]==nums[i]) continue;
            int sum2 = 0- nums[i];
            //search for another two;
            int left = i+1;
            int right = nums.length-1;
            while(left<right){
                int tmp = nums[left]+nums[right];
                if(tmp==sum2){
                    List<Integer> listone = new LinkedList<Integer>();
                    listone.add(nums[i]);
                    listone.add(nums[left]);
                    listone.add(nums[right]);
                    Collections.sort(listone);
                    res.add(listone);
                    left++;
                    right--;
                    while(left<right&&nums[left]==nums[left-1]){
                        left++;
                    }
                    while(left<right&&nums[right]==nums[right+1]){
                        right--;
                    }
                }
                else if(tmp < sum2){
                    left++;
                }else{
                    right--;
                }
            }
        }
        return res;
    }
}

 

问题非常清楚，这里需要注意的是：我们不能使用剪枝策略，而是应该遍历所有的没有重复元素的三元组。

public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        boolean flag=false;
        List<List<Integer> > res=new LinkedList<List<Integer> >();
        for(int i=0;i<nums.length;i++){
            if(flag) break;
            int num1=nums[i];
            if(i>0){
                if(num1==nums[i-1]){
                    continue;
                }
            }
            for(int j=i+1;j<nums.length;j++){
                if(flag) break;
                int num2=nums[j];
                if(j>i+1){
                    if(num2==nums[j-1]){
                        continue;
                    }
                }
                for(int k=j+1;k<nums.length;k++){
                    int num3=nums[k];
                    if(k>j+1){
                        if(num3==nums[k-1]){
                            continue;
                        }
                    }
                    int sum = num1 + num2 + num3;
                    /*
                    the pruning method is wrong, such as -2,-1,0,1,2,3
                    -2+2+3>0,how ever -1+0+1 is missed!!
                    if(sum>0){
                        flag=true;
                        break;
                    }
                    */
                    if(sum==0){
                        List<Integer> listone=new ArrayList<Integer>();
                        listone.add(num1);
                        listone.add(num2);
                        listone.add(num3);
                        res.add(listone);
                    }
                }
            }
        }
        return res;
    }
}

上面的注释处举例说明了为什么不能根据sum的大小进行剪枝的原因。