leetCode:三数之和

leetCode:三数之和

题目：

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

注意：答案中不可以包含重复的三元组。
示例 1：
输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

示例 2：
输入：nums = []
输出：[]

示例 3：
输入：nums = [0]
输出：[]

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

思路：

求三数之和为 0, nums[i] + nums[j] + nums[k] = 0 可以转换为求两个数之和，也即是 nums[i] + nums[j] = -nums[k]; 那么就可以通过指针移动去查找满足公式的数字，而为了方便控制指针的移动，前提是一组数字是有序的。
那么大概的解题思路如下:
对数组排序

• 定义三个指针 i, left, right; 指针 i 指向的是求和项，指针 left, right 分别指向数组的头部和尾部
• 当 -nums[i] = nums[left] + nums[right] 时满足公式，指针 left, right 继续向中部移动
• 当 -nums[i] > nums[left] + nums[right] 时指针 left 向中部移动
• 当 -nums[i] < nums[left] + nums[right] 时指针 righr 向中部移动
• 当指针 left，right 交叉时停止移动，指针 i 向下移动，指针 left, right 重置

代码：

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        Set<List<Integer>> list = new HashSet<>();
        for (int i = 0; i < nums.length; i++) {
            int left = i + 1;
            int right = nums.length - 1;
            while (left < right) {
                if (nums[left] + nums[right] == -nums[i]) {
                    List<Integer> data = new ArrayList<>();
                    data.add(nums[i]);
                    data.add(nums[left]);
                    data.add(nums[right]);
                    list.add(data);
                    left++;
                    right--;
                } else if (nums[left] + nums[right] > -nums[i]) {
                    right--;
                } else {
                    left++;
                }
            }
        }
        return new ArrayList<>(list);
    }
}