Leetcode--Java--15. 三数之和

题目描述

给你一个包含 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 = []
输出：[]

思路

1. 双指针。 双指针需要有序，先将数组排序，保证指向的值 i < j < k
   
2. 固定i，双指针枚举k，j ,如果j增大，k必然减小，因为 i + j + k = 0
3. 去重。若当前数和上一个数是一样的话，就跳过。 因为上一个已经把取nums[i]的情况枚举完了

优化后的方法：

1. 固定i，动态调整另外两个指针
2. i指针要去重。同时在加入到将结果集后l和r指针也要去重，改变指针的位置指向下一个不同的地方

代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
       List<List<Integer>> res = new ArrayList<>();
       //先排序
       Arrays.sort(nums);
       int len = nums.length;
       for (int i = 0; i < len; i ++ ) {
           //去重,如果和上一个i指向的数一样
           if (i > 0 && nums[i] == nums[i - 1]) continue;
           //j k两个指针不能相遇
           for (int j = i + 1, k = len - 1; j < k; j ++ ) {
               //去重，如果和上一个j指向的数相同
               if (j > i + 1 && nums[j] == nums[j - 1]) continue;
               //找到满足 >= 0的最小的 k
        //试探法。 如果下一个数满足，就用下一个数。使得退出while循环时k不会等于j
               while (j < k - 1 && nums[i] + nums[j] + nums[k - 1] >= 0) k --;
                if (nums[i] + nums[j] + nums[k] == 0) {
                    // List<Integer> ans = new ArrayList<>();
                    // ans.add(nums[i]);
                    // ans.add(nums[j]);
                    // ans.add(nums[k]);
        res.add(new ArrayList<Integer>(Arrays.asList(nums[k], nums[i], nums[j])));
                    // res.add(ans);
                }
           }
       }
       return res;
    }
}

优化后的方法

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
      Arrays.sort(nums);
      int len = nums.length;
      List<List<Integer>> res = new ArrayList<>();
      for (int i = 0; i < len - 2; i ++ ) {
          if (nums[i] > 0) break; //如果i指的数大于0，则三个数一定大于0
          //去重
          if (i > 0 && nums[i] == nums[i - 1]) continue;
          int l = i + 1, r = len - 1;
          while (l < r) {
              int s = nums[i] + nums[l] + nums[r];
               if ( s == 0){
                  res.add(new ArrayList(Arrays.asList(nums[i], nums[l], nums[r])));
                 //去重
                    while(l < r && nums[l] == nums[l + 1]) l ++;
                    while(l < r && nums[r] == nums[r - 1]) r --;
                    //上面退出循环时不等于下一项，但是这项已经被加入到结果集了
                    //所以得再次移动
                    l ++;
                    r --;
               }
            else  if (s < 0)
              l ++;
              else if (s > 0)
              r --;
          }
           }
    return res;
    }
}