天天看点

力扣 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=0开始,L指针指向i+1,R指针指向l-1(l为数组长度),判断nums[L]+nums[R]是否等于-nums[i]. 如果大于,则R指针减一,小于则L指针加一,相等就存于set中。set用于去重。

代码:(很慢)

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
          sort(nums.begin(),nums.end());
          int l=nums.size();
          set<vector<int>>se;
          for(int i=0;i<l-2;i++)
          {
              if(nums[i]>0)
                break;
              int target=-nums[i];
              int j=i+1;
              int k=l-1;
              while(j<k)
              {
                  auto sum=nums[j]+nums[k];
                  if(target==sum)
                  {
                      se.insert({nums[i],nums[j],nums[k]});
                      j++;
                      k--;
                  }
                  else if(sum>target)
                  {
                      k--;
                  }
                  else if(sum<target)
                  {
                      j++;
                  }
              }
          }
          return vector<vector<int>>(se.begin(), se.end());
    }
};
           

继续阅读