15. 3Sum

15. 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: 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]
]

/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
    
    //和two sum的解法一样，只不过多个循环，还有一点，注意要去重。  
    var len = nums.length;
    
    var res = [];

    
    //这里有个坑，就是sort函数对于负数，是根据绝对值排的，不行你们试试 [-4,-2,-2]变成了[-2,-2,-4];
    //所以这里我加了个callback
    nums.sort(function(a,b){ return a-b;});
    
    for(var i = 0;i<len;i++){
        
        //这就变成two sums问题
        var target = 0 - nums[i];
        
        var left = i + 1; var right = len - 1;
        
        
        //去重  比如   S = [-1, 0, 1, 2, -1, -4]  排序后变成 [-4,-1,-1,0, 1, 2]
            
        //第一种  i--> -4 j--> -1 k一直遍历都不行
            
        //第二种  i--> -4 j--> -1(这里其实往前走了一位的) j从0开始到2， 这种其实和第一遍一样，我们要去掉这种重复的。     
        
        //第三种  假如说 -1 0 1 已经是一组解了， 然后由于-1有2个数，这就造成重复解
        
        
        //双指针解法
        while(left < right){
                    
            var l = nums[left];
            var r = nums[right];
            var temp = l + r;
        
            //首先要保证一个解
             if(temp == target){
                 var subres = [];
                 subres.push(nums[i]);
                 subres.push(l);
                 subres.push(r);
                 
                 res.push(subres);
                               
                //去重
                while(nums[left] == nums[left+1]){
                
                      left++;
              
                }
            
                while(nums[right] == nums[right-1]){
                
                     right--;
              
                }
                 
                 left++;
                 right--; 
                 
                
             }else if(temp < target){
                 
                 left++;
                 
             }else{
                 
                 right--;
             } 
              
        }
        
        //去重
        while(i + 1 < len && nums[i] == nums[i+1]){
            
            i++;
        }
                        
    }
    return res;
};