leetcode 15. 三数之和

15. 三数之和

难度中等4721收藏分享切换为英文接收动态反馈

给你一个包含 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

简单一句话，先固定第一个数a，然后b、c只能从两边向中间靠（在a之后）。细节条件就是去重处理

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
       Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        Set<List<Integer>> set = new HashSet<>();
if (nums == null || nums.length < 3)
        return new ArrayList<>();

        for (int i = 0; i < nums.length; i++) {

            if (nums[i] > 0) {
                break;
            }

            int l = i + 1;
            int h = nums.length - 1;
            int sum = 0;

            while (l < h) {
                sum = nums[i] + nums[l] + nums[h];
                if (sum < 0) {
                    l++;
                } 
                if (sum > 0) {
                    h--;
                }
                if (sum == 0) {
                    List<Integer> list = new ArrayList<>();
                    list.add(nums[i]);
                    list.add(nums[l]);
                    list.add(nums[h]);
                    if (!set.contains(list)) {
                        res.add(list);
                        set.add(list);
                    }
                    h--;
                l++;
                }
            }
        }

        return res;

    }
}