【数组】n数之和问题

https://leetcode-cn.com/problems/3sum/solution/san-shu-zhi-he-by-leetcode-solution/

 

https://leetcode-cn.com/problems/4sum/solution/si-shu-zhi-he-by-leetcode-solution/

 

15. 三数之和

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

 

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

 

 

 

示例：

 

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

 

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

 

解答：

方法一：排序+双指针

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        
        //排序，以使枚举时不出现重复的三元组
        Arrays.sort(nums);
        int n = nums.length;
        for(int first = 0;first<n;first++){

            //需要和上次枚举的数不同
            if(first>0&&nums[first] == nums[first-1]){
                continue;
            }
            int target = 0 - nums[first];

            for(int second = first + 1;second<n;second++){

                //需要和上次枚举的数不同
                if(second>first+1&&nums[second]==nums[second-1]){
                    continue;
                }
                int third = n-1;
                
                while(second<third&&nums[second]+nums[third]>target){
                    third--;
                }

                // 如果指针重合，随着 b 后续的增加
                // 就不会有满足 a+b+c=0 并且 b<c 的 c 了，可以退出循环
                if(second == third){
                    break;
                }
                if(nums[second]+nums[third] == target){
                    List<Integer> list = new ArrayList<>();
                    list.add(nums[first]);
                    list.add(nums[second]);
                    list.add(nums[third]);
                    res.add(list);
                }
            }
        }

        return res;
        
    }
}