Leetcode015. 3sum

题目

• 原题链接：https://leetcode.com/problems/3sum
• 题目内容：

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

Note:

The solution set must not contain duplicate triplets.

解题思路

首先对数组nums排序，其次得到的结果要去重，避免出现重复的情况。
时间复杂度O(N^2)，空间复杂度O(N)。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ret;
        sort(nums.begin(), nums.end());
        for(int i=0; i<nums.size(); ++i){
            int j=i+1, k=nums.size() -1;
            if(i&&nums[i] == nums[i-1])
                continue;
            while(j<k){
            if(nums[j]+nums[k] == -nums[i]){
                vector<int> temp;
                temp.push_back(nums[i]);
                temp.push_back(nums[j]);
                temp.push_back(nums[k]);
                ret.push_back(temp);
                ++j;
                --k;
            }
            else if(nums[j]+nums[k] < -nums[i])
                ++j;
            else
                --k;
            }
    }
    auto pos = unique(ret.begin(),ret.end());
    ret.erase(pos,ret.end());
    return ret;
    }
};

知识点

• unique函数
  -unique的作用是从输入序列中“删除”所有相邻的重复元素。 unique去除相邻的重复元素(只保留一个),其实它并不真正把重复的元素删除，是把重复的元素移到后面去了，然后依然保存到了原数组中，然后 返回去重后最后一个元素的地址，因为unique去除的是相邻的重复元素，所以一般用之前都会要排一下序。
• sort函数
  sort(vec.begin(), vec.end())默认是升序排列。
• erase
  ret.erase(pos) 删除迭代器pos所指向的元素，返回一个迭代器，它指向被删除元素后面的元素。如果p指向容器内的最后一个元素，则返回的迭代器指向容器的超出末端的下一位置，如果p本身就是指向超出末端的下一位置的迭代器，则该函数未定义。