刷题笔记-三数之和

三数之和

中等难度

解决方案

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector< vector<int> > vec;
        sort(nums.begin(), nums.end());
        for(int i=0; i<nums.size(); i++)
        {
      
     
            int j=i+1, k=nums.size()-1;
            if(nums[i]>0 || i>nums.size()-3)  //最小的num大于0或还未遍历的num少于3个
                return vec;
            if(i>0 && nums[i]==nums[i-1])
                continue;
            while(j<k)
            {
                if(nums[i]+nums[j]+nums[k]>0)
                {
                    k--;
                    while(k>=0 && nums[k]==nums[k+1])
                    {k--;}
                }
                else if(nums[i]+nums[j]+nums[k]<0)
                {
                    j++;
                    while(j<nums.size() && nums[j]==nums[j-1])
                    {j++;}
                }
                else{
                    vector<int> temp;
                    temp.push_back(nums[i]);
                    temp.push_back(nums[j]);
                    temp.push_back(nums[k]);
                    vec.push_back(temp);
                    j++; k--;
                    while(k>=0 && nums[k]==nums[k+1])
                    {k--;}
                    while(j<nums.size() && nums[j]==nums[j-1])
                    {j++;}
                }
            }
        }
        return vec;
    }
};

一份别人的代码

虽然时间复杂度是O(N^2)，运行超时，但有可取之处。比如对set容器的运用。

class Solution {
public:
    static bool cmp(const int&a,const int &b)
    {
        return a<b;
    }
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end(),cmp);
        vector<int>::iterator it;
        vector<vector<int>> res;
        set<vector<int>> a;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]>0) break;
            for(int j=i+1;j<nums.size();j++)
            {
                it=find(nums.begin()+j+1,nums.end(),0-nums[i]-nums[j]);  //重点,find用法
                if(it!=nums.end())
                {
                    vector<int> temp;
                    temp.push_back(nums[i]);
                    temp.push_back(nums[j]);
                    temp.push_back(*it); //等效于 temp.push_back({nums[i],nums[j],*it});
                    a.insert(temp); //重点
                }
            }
        }
        for(auto k:a)
        {
            res.push_back(k);
        }
        return res;
    }
};

知识点

双指针算法应用
set容器，特点是元素具有唯一性

吐槽

按照这刷题进度和出现bug的概率，怕是药丸。

注意

• 严防数组的下标越界溢出
• if 或者 while (条件1 && 条件2) 如果条件1为false，就不会进行条件2的判断