15. 三数之和

15. 三数之和

题目

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

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

示例

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

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

分析

这个题目的难点在于去重，之前我的做法是：

先对于每两个数，求他们的和，存在map里，全部做完之后，遍历原始数组，对于每个数c，如果-c存在于map中，说明就找到三个满足的数。

上面这样做，可以得到答案，去重比较麻烦，写起来比较繁琐，而且最终超时。

参考的答案

链接：https://leetcode-cn.com/problems/3sum/solution/pai-xu-shuang-zhi-zhen-zhu-xing-jie-shi-python3-by/

代码

    public static List<List<Integer>> threeSum(int[] nums) {
        int n = nums.length;
        Arrays.sort(nums);
        List<List<Integer>> R = new ArrayList<>();
        for (int i = 0; i < n - 2; i++) {
            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) {
                if (nums[i] + nums[l] + nums[r] == 0) {
                    List<Integer> t = new ArrayList<>();
                    t.add(nums[i]);
                    t.add(nums[l]);
                    t.add(nums[r]);
                    R.add(t);
                    while (l < r && nums[l] == nums[l + 1]) {
                        l++;
                    }
                    while (l < r && nums[r] == nums[r - 1]) {
                        r--;
                    }
                    l++;
                    r--;
                } else if (nums[i] + nums[l] + nums[r] > 0) {
                    r--;
                } else {
                    l++;
                }
            }
        }
        return R;
    }