【算法笔记】力扣热题100（LeetCode hot-100） 15: 三数之和

力扣热题100（LeetCode hot-100）之 15: 三数之和

本文主要记录算法思路，着急要答案的同学可以直接跳转到最后的代码。

题目

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，
同时还满足 nums[i] + nums[j] + nums[k] == 0 。
请你返回所有和为 0 且不重复的三元组。

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

示例 1：
输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：
输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：
输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。
 

提示：
3 <= nums.length <= 3000
-105 <= nums[i] <= 105

题解

求什么？

根据题目描述，我们需要找出所有和为 0 且不重复的三元组。
通俗的说，就是找出数组中所有的三个数，使得这三个数的和为0，找出所有的组合。

求这个问题，一般角度有哪些解法？

思路：

我们可以使用排序和双指针法来解决这个问题。

1. 首先将数组排序；

2. 遍历数组，对于每个元素，双指针求剩余元素中寻找两个数，使得三数之和为 0；

3. 注意跳过重复元素，避免重复结果。

   3.1 外层循环遍历时，如果当前元素与上一个元素相同。（向前对比重复元素）

   3.2 内层获得结果后，往后移动指针，跳过重复元素。（向后对比重复元素）

例如：

输入：我有一个数组 [-1,0,1,2,-1,-4]。

过程：

1. 将数组排序，得到 [-4,-1,-1,0,1,2]。
2. 遍历数组，对于每个元素，使用双指针在剩余元素中寻找两个数，使得三数之和为 0。
3. 如果找到满足条件的三个数，将其加入结果集。
4. 由于题目要求结果集中不能包含重复的三元组，因此在外层便利时，需要跳过重复的元素。即若当前元素与上一个元素相同，则跳过。

答案

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        int len = nums.length;
        // 先排序
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < len - 1; i++) {
            // 如果当前元素大于0，那么后面的元素肯定大于0，不可能和为0
            if (nums[i] > 0) {
                break;
            }
            // 跳过重复元素
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            int target = -nums[i];// 另外两个数的和
            int left = i + 1;
            int right = len - 1;
            while (left < right) {
                int sum = nums[left] + nums[right];
                if (sum == target) {
                    res.add(List.of(nums[i], nums[left], nums[right]));
                    // 跳过重复元素
                    while (left < right && nums[left + 1] == nums[left]) {
                        left++;
                    }
                    left++;
                } else if (sum < target) {
                    left++;
                } else {
                    right--;
                }
            }
        }
        return res;
    }
}