美文网首页
15. 三数之和

15. 三数之和

作者: 一直流浪 | 来源:发表于2022-10-04 13:17 被阅读0次

15. 三数之和

难度:中等

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

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

示例:

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

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

解法一:双指针法

分析这道题,我们一共需要解决两个问题,一个是找到所有符合条件的数组,第二个是如何去除重复的数组。

首先我们需要先对数组进行排序,我们可以用变量i遍历数组,nums[i]就是第一个元素,我们可以使用两个指针l和r依次指向i+1和len-1,其中len是nums数组的长度。

  • 当i < len 循环i,若nums[i]>0,直接return。因为当nums[i]>0的话,往后面就没有符合条件的。

    • 当l<r,就一种进行下面循环:
      • 我们通过计算t = nums[i] + nums[l]+nums[r],
        • 若t=0,我们就直接他们加入最终结果集;
        • 若t>0,说明是nums[r]太大了,我们应该将r指针向右移动;
        • 若t<0,说明是nums[l]太小了,我们应该将l指针向左移动。
  • 这道题还有一个难点,就是如何去除重复的结果?

    • 我们已经对数组进行了排序,所以如果有重复的结果,那么重复结果的下标i或l或r一定是相邻的,所以我们只需要在找到结果的时候,
      * 判断nums[l+1]是否等于nums[l],num[r-1]是否等于nums[r]
      * 若等于,我们直接移动指针,直到不相等。
      • 还有就是在移动i的时候,我们也需要判断,
        * 如果nums[i]和nums[i-1]相等,我们就可以直接跳过这次循环,进行下一次循环。

代码:

    public static List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);
        int len = nums.length;      
        for(int i =0;i<len;i++) {
            if(nums[i]>0) {
                return result;
            }
            
            if(i>0 && nums[i] == nums[i-1]) {
                continue;
            }
            int l = i+1;
            int r = len -1;     
            while (l<r) {
                int t = nums[i] + nums[l]+nums[r];
                if(t == 0) {
                    List<Integer> list = new ArrayList<>();
                    list.add(nums[i]);
                    list.add(nums[l]);
                    list.add(nums[r]);
                    result.add(list);   
                    
                    while(l < r && nums[l+1] == nums[l]) {
                        l++;
                    }
                    while(l<r && nums[r-1] == nums[r]) {
                        r--;
                    }
                    ++l;
                    --r;
                    
                }else if(t > 0) {
                    r--;
                }else {
                    l++;
                }               
            }
        }
        return result;
    }

相关文章

网友评论

      本文标题:15. 三数之和

      本文链接:https://www.haomeiwen.com/subject/eiiqwrtx.html