牛客网：三数之和

 

目录

方法一：暴力

方法二：双指针

---

方法一：暴力

三重循环应该大家都能想到

但是这个过不了啊

怎么办呢，就利用一个O(nlogn)复杂度的排序来降低一下复杂度，怎么降低呢？排序之后相同元素会在一起，我们遇到相同的直接跳过就行了。

代码如下所示：

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        vector<vector<int>> res;
        if(num.size()<3) return res;
        sort(num.begin(), num.end());
        for(int i=0;i<num.size()-2;++i){
            if(i&&num[i]==num[i-1]) continue;
            for(int j=i+1;j<num.size()-1;++j){
                if(j>i+1&&num[j]==num[j-1]) continue;
                for(int k=j+1;k<num.size();++k){
                    if(k>j+1&&num[k]==num[k-1]) continue;
                    if(num[i]+num[j]+num[k]==0){
                        res.push_back({num[i],num[j],num[k]});
                    }
                }
            }
        }
        return res;
    }
};

方法二：双指针

我们可以利用双指针的方法把时间复杂度降到O(n^2)，首先还是排序，因为这可以让我们跳过重复元素。其次这也方便我们使用双指针。

接下来对数组中的数进行如下处理：

1.设置后面两个数的指针，即后面区域的左右指针

• 如果三个数相加等于0，则将三个数放入答案，然后更新两个指针至不重复元素位置
• 如果大于0，将右指针往左移一位
• 如果小于0，将左指针往右移一位

退出的条件就是当左指针的位置>=右指针的位置时

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        vector<vector<int>> res;
        if(num.size()<3) return res;
        sort(num.begin(), num.end());
        for(int i=0;i<num.size();++i){
            if(i&&num[i]==num[i-1]) continue;
            int l=i+1,r=num.size()-1;
            while(l<r){
                if(num[i]+num[l]+num[r]==0){
                    res.push_back({num[i],num[l],num[r]});
                    while(l+1<r&&num[l]==num[l+1]) ++l;
                    while(r-1>l&&num[r]==num[r-1]) --r;
                    ++l,--r;
                }else if(num[l]+num[r]+num[i]>0){
                    --r;
                }else{
                    ++l;
                }
            }
        }
        return res;
    }
};