[Lecode]15. 三数之和

2021/4/8,开始刷Lecood TOP100,题意如下:
[Lecode]15. 三数之和_第1张图片
思路就是先排序,然后从左开始先固定一个值,然后在使用两个下标,在剩下的里面找看有没有能够和为0的,有的话就存下来。要注意的细节就是相等的值要跳过避免重复。
c++参考代码:

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
     vector<vector<int>> ans;
    if(nums.size()<3)
    return ans;
    sort(nums.begin(),nums.end());
    for(int i=0;i<nums.size();i++)
    {
        if(nums[i]>0)//左边的值大于0就不可能和为0了。
        return ans; 
        if(i>0&&nums[i]==nums[i-1])
        continue;
        int l=i+1,r=nums.size()-1;
        while(l<r)
        {
            int now=nums[i]+nums[l]+nums[r];
            if(now>0)//大于0说明右边界大了,往左缩。
            {
                r--;
            }
            else if(now<0)//小于0说明左边界小了,往右缩。
            {
                l++;
            }
            else
            {
                ans.push_back(vector<int>{nums[i],nums[l],nums[r]});
                l++;
                r--;
                while(l<r&&nums[l]==nums[l-1])//避免重复
                  l++;
                while(l<r&&nums[r]==nums[r+1])
                  r--;
            }
        }
    }
     return ans;
    }
};

[Lecode]15. 三数之和_第2张图片
时间复杂度O(N2),空间复杂度O(N)。

你可能感兴趣的:(菜就多做题,c++,leetcode)