leetcode:三数之和---双指针

news2024/9/23 9:32:50

问题：

给你一个整数数组 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

代码（思路在代码上注释了）：

 public List<List<Integer>> threeSum(int[] nums) {
        //排序，方便后续双指针移动
        Arrays.sort(nums);
        //返回值类型
        List<List<Integer>>lists=new ArrayList<>();
        //定义集合里面的集合，数组内容将会进入的集合
        List<Integer>list;
        //数组的长度
        int len=nums.length;
        //后续使用的三个指针，其中a为定。b,c为动，sum为三个数之和
        // a，b，c同时为数组索引
        int a,b,c,sum;
        //循环遍历
        for(int i=0;i<len-2;i++){
            //a,b,c赋初位置
            a=i;b=i+1;c=len-1;
            if(nums[a]>0){
                //如果nums[a]大于0那么再加两个大于0的数还是大于0
                continue;
            }
            if(a>0&&nums[a-1]==nums[a]){
                //当a走到下一个时避免与上一个相同
                continue;
            }
            while(b<c){
                //求和
                sum=nums[a]+nums[b]+nums[c];
                if(sum>0){
                    //大于0，则num[c]过大c--;
                    //同时避免相同
                    //此处建议这样写，c--单独写会超时
                    while(b<c&&nums[c]==nums[--c]);
                }else if(sum<0){
                    //小于0，则num[b]过小b++;
                    //同时避免相同
                    //此处建议这样写，b++单独写会超时
                    while(b<c&&nums[b]==nums[++b]);
                }else{
                    //sum=0时，加入内层集合
                    //注意每次创建新的内部集合（List<List<Integer>>此处是泛型）
                    //就当作内部和外部（不是准确这种说法）
                    list=new ArrayList<>();
                    list.add(nums[a]);
                    list.add(nums[b]);
                    list.add(nums[c]);
                    //将内部集合放入外层集合
                    //注意外层集合创建一次即可
                    lists.add(list);
                    //此处是为了避免出现相同的内部集合
                    while(b<c&&nums[b]==nums[++b]);
                    while(b<c&&nums[c]==nums[--c]);
                }
            }
        }
        //最终返回外层集合
        return lists;
    }