Leetcode15 三数之和

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

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

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

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

解法:首先对数组排序,然后设定两个指针b,c,三数之和符合要求即可保存到结果vector中。

           注意控制别越界就好,还有注意跳过数组中的重复内容,可以减少时间复杂度!

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
      vector <vector<int>> result;  
      if (nums.size()<3) 
        return result;
      sort(nums.begin(),nums.end());
      int temp = 0;
      int a , b , c; //控制位置指针
      for (int a=0;a<nums.size();a++)
      {
        if (nums[a]>0) break;
        if (a!=0 && nums[a]==nums[a-1])  continue;//跳过重复
        b = a+1;
        c = nums.size()-1;
        while(b<c)//区间中寻找
        {
          int sum = nums[a]+nums[b]+nums[c];
         if (sum<temp) //小于目标
          {
            b++;
          }
          else if (sum>temp) //大于目标
          {
            c--;
          }
          else//找i到目标
          {
          result.push_back({nums[a],nums[b],nums[c]});
          while(b<c && nums[b]==nums[b+1]) b++;//跳过重复
          while(b<c && nums[c]==nums[c-1]) c--;//跳过重复
          b++;
          c--;
          }
        }
      }
        return result;
    }
};

猜你喜欢

转载自blog.csdn.net/u010760034/article/details/83274662