力扣刷题日志--- 15 三数之和

题目描述

给定一个包含 n 个整数的数组 nums，
判断 nums 中是否存在三个元素 a，b，c ，
使得 a + b + c = 0 ？找出所有满
足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[  [-1, 0, 1],
    [-1, -1, 2]
 ]

三指针法

vector<vector<int>> threeSum(vector<int>& nums) {
  sort(nums.begin(),nums.end());//排序
  if(nums.size() <= 2||nums.front() > 0||nums.back() < 0)
  return {};
  vector<vector<int> > ans;
  int fix,i,j;
  for(fix = 0;fix < nums.size();fix++)//固定指针，遍历数组
  {
     if(fix > 0&&nums[fix] == nums[fix - 1]) continue;//防止固定指针重复
     if(nums[fix] > 0) break;//如果固定指针指向正数，就停止，因为此时三数字之和一定为正数
     i = fix + 1;//左指针从固定指针右侧开始遍历
     j = nums.size() - 1;//右指针从数组最右侧开始
     while(i < j)
     {
       int s = nums[fix] + nums[i] + nums[j];
       if(s > 0) j--;
       if(s < 0) i++;
       if(s == 0&&i < j)
       { 
         ans.push_back({nums[fix],nums[i],nums[j]});
         while(i < j&&nums[i] == nums[++i]);//i前进至下一个不重复的数，防止重复
         while(i < j&&nums[j] == nums[--j]);//j退至后一个不重复的数
       }
     }
  }
  return ans;
}

核心：数组排序后可用双指针查找两数之和，如果和大于查找值，右指针后退一步（或直至下一个不重复的数），如果和小于查找值，左指针前进一步