LeetCode15.三数之和

LeetCode15.三数之和

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

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

示例：

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

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
题解：
java版本：

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

/**
 * 难点：如何去除重复解
 * 1.对于数组长度n，如果数组为null或者数组长度小于3，返回[]
 * 2.对数组排序
 * 3.遍历排序后的数组
 * （1）. 如果nums[i]>0:因为已经是排好序，所以后面不可能有三个树加和等于0，直接返回结果
 * （2）. 对于重复元素：跳过，避免出现重复解
 * (3). 令左指针 L=i+1，右指针 R=n-1，当L<R时，执行循环：
 * a.当nums[i]+nums[L]+nums[R]==0，执行循环，判断左和右是否和下一位置重复，去除重复解。并同时将L,R 移到下一个位置，
 * b.若和大于0，说明nums[R]太大，R左移
 * c.若和小于0，说明nums[L]太小吗，L右移l
 **/
public class leetcode15_ThreeSum {
    public static void main(String[] args) {
        int[] nums = new int[6];//-1,0,1,2,-1,-4
        nums[0] = -1;
        nums[1] = 0;
        nums[2] = 1;
        nums[3] = -2;
        nums[4] = -1;
        nums[5] = -4;
        System.out.println(threeSum(nums));
    }

    public static List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> lists = new ArrayList<>();
        Arrays.sort(nums);//对数组nums进行排序
        int len = nums.length;
        if (nums == null || len < 3) return lists;//判断数组是否为空或者长度小于3
        for (int i = 0; i < len; i++) {
            if (nums[i] > 0) break;//如果当前数组大于哦，那么三叔之后一定大于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) {
                    lists.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 lists;
    }
}