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]

]

 

 

先给数组排序，判断第一个数字是否大于零或者最后一个数字是否小于零，遍历数组，跳出条件有当前数字已经大于零，因为我们是用当前数字加上后面的数字取和的，第一个数字都大于零了，所以后面也就不满足条件了，还有就是判断当前数字和下一位数字是否相等，因为题目要求不能重复，遍历后面的元素之和是否满足当前数字的相反数，满足的话就添加到结果集合，然后第二个下标右移，第三个下标左移，当然相等的话就需要继续排除，不满足的话判断是小于还是大于，分别移动第二个下标和第三个下标。

 

public static List<List<Integer>> threeSum(int[] nums) {
    List<List<Integer>> res = new ArrayList<>();
    Arrays.sort(nums);
    if (nums.length == 0 || nums[0] > 0 || nums[nums.length - 1] < 0) return res;
    for (int k = 0; k < nums.length; k++) {
        if (nums[k] > 0) break;
        if (k > 0 && nums[k] == nums[k - 1]) continue;
        int target = 0 - nums[k];
        int i = k + 1, j = nums.length - 1;
        while (i < j) {
            if (nums[i] + nums[j] == target) {
                res.add(Arrays.asList(nums[k], nums[i], nums[j]));
                while (i < j && nums[i] == nums[i + 1]) ++i;
                while (i < j && nums[j] == nums[j - 1]) --j;
                ++i; --j;
            } else if (nums[i] + nums[j] < target) ++i;
            else --j;
        }
    }
    return res;
}