三个数之和

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

/**
 *
 * 给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，
 * 使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。
 *
 * 注意：答案中不可以包含重复的三元组。
 *
 * 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
 * 满足要求的三元组集合为：
 * [
 *   [-1, 0, 1],
 *   [-1, -1, 2]
 * ]
 *
 * 解题思路：
 *      首先我们对数组进行排序，然后定义两个指针，
 *      分别指向 当前遍历索引的 下一个 和 数组 最后一个元素的索引位置
 *      在计算的过程当中，我们需要防止重复数字的重复计算。
 */
public class ThreeSum15 {

    public List<List<Integer>> threeSum(int[] nums) {
        //定义返回值
        List<List<Integer>> res = new ArrayList<>();
        int n = nums.length;
        //如果传入数据长度小于 3 直接 返回，题意要求三个数之和
        if(n < 3) return res;
        //先进行排序操作
        Arrays.sort(nums);

        //三个数的和
        int sum = 0;
        //遍历整个数组
        for(int i = 0; i < n; i ++){
            //如果当前这个数 num > 0 循环结束
            if(nums[i] > 0) break;
            //防止相同数字重复计算，造成结果相同
            if(i > 0 && nums[i] == nums[i - 1]) continue;
            //定义左右边界
            int l = i + 1;
            int r = n - 1;
            while(l < r){
                //计算三个数的和
                sum = nums[i] + nums[l] + nums[r];
                if(sum == 0){
                    //满足条件，装入集合
                    res.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 {
                    r --;
                }
            }
        }
        return res;
    }

    public static void main(String[] args) {
        ThreeSum15 t = new ThreeSum15();
        int[] nums = {-1, 0, 1, 2, -1, -4};
        List<List<Integer>> lists = t.threeSum(nums);
        System.out.println(lists);
    }
}

题目来源：力扣官方