leetcode+three sum+双指针

点击打开链接

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string>
#include<cstring>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<vector>
#include<set>
#include<limits.h>
#include<assert.h>
#include<queue>
#define LL long long
#define inf 0x3f3f3f3f
#define mod 1e9+7
using namespace std;
//首先对原数组进行排序，然后开始遍历排序后的数组，这里注意不是遍历到最后一个停止，而是到倒数第三个就可以了，
//然后我们还要加上重复就跳过的处理，对于遍历到的数，我们用0减去这个数得到一个sum，我们只需要再之后找到两个数之和等于sum即可，
//这样一来问题又转化为了求two sum，这时候我们一次扫描，找到了等于sum的两数后，加上当前遍历到的数字，按顺序存入结果中即可
//然后还要注意跳过重复数字
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        sort(nums.begin(),nums.end());
        for(int k=0;k<nums.size();k++){
            if(nums[k]>0) break;
            if(k>0&&nums[k]==nums[k-1]) continue;// 为的是不能得到重复的结果
            int target=0-nums[k];
            int i=k+1,j=nums.size()-1;
            while (i<j) {
                if(nums[i]+nums[j]==target){
                    res.push_back({nums[k],nums[i],nums[j]});//push进去是{
                    while (i<j&&nums[i]==nums[i+1])++i;//为的是不能得到重复的结果
                    while(i<j&&nums[j]==nums[j-1])--j;//为的是不能得到重复的结果
                    ++i; --j;
                }else if(nums[i]+nums[j]<target)++i;//比较小，所以i移动
                else --j; //比较大j移动
            }
        }
        return  res;
    }
};
int main()
{
    Solution temple;
    vector<int>Q;
    Q.push_back(-1);  Q.push_back(0); Q.push_back(1); Q.push_back(2); Q.push_back(-1); Q.push_back(4);
    return 0;
}