leetcode 三数之和

题目链接：https://leetcode-cn.com/problems/3sum/
总结知识点：

• 题目中要求a+b+c == 0，如果使用朴素的方法时间复杂度为$n^3$
• 变换一下等式为 a+b=-c，那么时间复杂度为$n^2$
• 具体方法为首先将数组排序，然后使用双指针方法，计算num[l]+num[r]+num[i]的数值，如果为0则加入最后答案数组，如果大于0则说明正数比较大，我们让r–，同理如果小于0，我们让l++，使用set去重，需要考虑有一堆重复数字的情况，否则会超时，在跳跃重复数字的时候需要关心下标是否越界。
• 加油！

#include <iostream>
#include <string>
#include <vector>
#include <set>
#include <queue>
using namespace std;
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        int numsLen = nums.size();
        set<vector<int> > ans;
        for(int i = 0; i < numsLen; i++) {
            int l = 0, r = numsLen - 1;
            while(l < r) {
                if(l >= i || r <= i) break;
                if(nums[l] + nums[r] + nums[i] == 0) {
                    vector<int> v{nums[l], nums[i], nums[r]};
                    ans.insert(v);
                    while(l + 1 < numsLen && nums[l] == nums[l + 1]) l++;
                    while(r > 0 && nums[r] == nums[r - 1]) r--;
                    l++, r--;
                } else if(nums[l] + nums[r] + nums[i] > 0) {
                    while(r > 0 && nums[r] == nums[r - 1]) r--;
                    r--;
                } else if(nums[l] + nums[r] + nums[i] < 0) {
                    while(l + 1 < numsLen && nums[l] == nums[l + 1]) l++;
                    l++;
                }
            }
        }
        vector<vector<int> > res;
        for(auto it = ans.begin(); it != ans.end(); it++) {
            res.push_back(*it);
        }
        return res;
    }
};

int main() {
    Solution s;
    vector<int> v{0,0,0};

    auto ans = s.threeSum(v);
    int len = unique(ans.begin(), ans.end()) - ans.begin();
    for(int i = 0; i < len; i++) {
        for(int j = 0; j < ans[i].size(); j++) {
            cout << ans[i][j] << " ";
        }
        cout << endl;
    }

    return 0;
}