leetcode刷题-[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 = []
输出：[]

示例 3：

输入：nums = [0]
输出：[]

思路

将三数之和固定首位的一个数看做target，就转换为了和两数之和；

由于题目中不可以包含重复的三元组，所以要对数组排序并判断；

1. sorted的首位不能大于0 ，因为最小的首位都大于0了，肯定三数相加不为0
2. 不能包含重复数据，所以要将重复数字跳过
3. 固定首位后，维护 左指针 i + 1 和 右指针 num.length - 1
   • 相加等于target。放入结果集，更新左指针和右指针，注意循环条件和重复数字的跳过
   • 小于target 说明应该移动左指针了
   • 否则移动右指针
4. 将下一位固定首位target。重复2，3步骤，直到数据遍历完成

关键代码

public List<List<Integer>> threeSum(int[] nums) {
  List<List<Integer>> ans = new ArrayList<>();
  if (nums == null || nums.length < 3) return ans;

  // -4, -1, -1, 0, 1, 2
  Arrays.sort(nums);

  // 如果第一个数大于0 ，后面的都不成立了
  if (nums[0] > 0) return ans;

  for (int i = 0; i < nums.length - 2; i++) {
    // 枚举a排除重复
    if (i > 0 && nums[i - 1] == nums[i]) continue;

    int target = -nums[i];

    // 第二个元素的指针
    int j = i + 1;
    // 第三个元素的指针
    int k = nums.length - 1;

    while (j < k) {
      // 如果成立
      if (nums[j] + nums[k] == target) {
        ans.add(Arrays.asList(nums[i], nums[j], nums[k]));
        j++;
        k--;
        // 判断重复
        while (j < k && nums[j] == nums[j - 1]) j++; // 因为已经移动过j 和 k指针。所以j要和j-1判断，k要和k+1判断
        while (j < k && nums[k + 1] == nums[k]) k--;
      } else if (nums[j] + nums[k] < target) {
        j++;
      } else {
        k--;
      }
    }
  }
  return ans;
}