三数之和(Three Sum)

取自leetCode第15题，三数之和。

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]
链接：https://leetcode-cn.com/problems/3sum

其实刚开始看到这道题的时候，第一感觉是很简单，但是仔细思考又觉得无从下手，自己尝试着画图，写代码都没有成功写出来。下午看了几篇网上的解答以及leetCode上的解答，看得并不是非常理解。在leetCode上这道题的通过率也不高。

在地铁上思考了两数之和的问题及解答，对左右双指针的使用有了有点理解，于是晚上回来跟着别人的代码画图，尝试去理解这道题的处理过程。

可以看出，上图解题思路是固定数组一端，将L和R指针分别指向固定元素之后所有元素的两端。L指针处于i和R指针中间的位置，可以灵活移动，由于数组是有序的，当nums[i]+nums[L]+nums[R]之和小于0时，可以让L指针往右移动一位，继续计算并比较大小。在L指针移动的过程中，不能超过R指针的下标。如果超过则必须重新使得i往后移动一位，再固定住，L指针=i+1，R指指针依然是放在数组最后一位，继续重复上述过程。需要注意的是中间的重复元素，比如上图中的两个-1，在i移动过程中必须跳过重复的-1，以免得到重复的组合。

代码如下：

package com.study.algorithm.array;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @Auther: JeffSheng
 * @Date: 2019/8/21 19:39
 * @Description: 找出有序数组中三个数之和为0的组合
 */
public class ThreeSum {

    public static List<List<Integer>> getThreeSum(int[] nums){
        List<List<Integer>> ans = new ArrayList();
        Arrays.sort(nums); // 排序
        int len = nums.length;
        if(nums == null || len < 3) return ans;
        for (int i = 0; i < len ; i++) {
            if(nums[i] > 0) break; // 如果当前数字大于0，则三数之和一定大于0，所以结束循环
            if(i > 0 && nums[i] == nums[i-1]) continue; // 去重
            int L = i+1;
            int R = len-1;
            while(L < R){
                int sum = nums[i] + nums[L] + nums[R];
                if(sum == 0){
                    ans.add(Arrays.asList(nums[i],nums[L],nums[R]));
                    while (L<R && nums[L] == nums[L+1]) L++; // 去重
                    while (L<R && nums[R] == nums[R-1]) R--; // 去重
                    L++;
                    R--;
                }
                else if (sum < 0) L++;
                else if (sum > 0) R--;
            }
        }
        return ans;
    }



    public static void main(String[] args) {
        int [] arr = {-4,-1,-1,0,1,2};
        List<List<Integer>> res = getThreeSum(arr);
        System.out.println(res);
    }
}

当遇到一些不知如何入手的算法题时，可以看下别人的解题思路，如果看不懂的话，尝试画图来梳理思路，一步一步模拟逻辑处理。