leetcode刷题系列C++-3sum

来源:互联网 发布:塔夫斯大学 知乎 编辑:程序博客网 时间:2024/04/28 15:42

Given an array S of n integers, are there elements abc in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

  • Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
  • The solution set must not contain duplicate triplets.

    For example, given array S = {-1 0 1 2 -1 -4},    A solution set is:    (-1, 0, 1)    (-1, -1, 2)

Subscribe to see which companies asked this question


class Solution {public:    vector<vector<int>> threeSum(vector<int>& nums) {        vector<vector<int> > result;        if(nums.size() < 3)            return result;        sort(nums.begin(),nums.end());        const int target = 0;        for(auto a = nums.begin(); a < prev(nums.end(),2); ++a)        {            auto b = next(a);            if(a != nums.begin() && *a == *(prev(a)))                continue;            auto c = prev(nums.end());            while(b < c)            {                if(*a + *b + *c < target)                {                    ++b;                }                else if(*a + *b + *c > target)                {                    --c;                }                else                {                    result.push_back({*a,*b,*c});                    ++b;                    --c;                }            }        }        //sort(result.begin(),result.end());        result.erase(unique(result.begin(),result.end()),result.end());        return result;    }};


0 0