Leetcode015. 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本身就是指向超出末端的下一位置的迭代器,则该函数未定义。

猜你喜欢

转载自blog.csdn.net/chenguiyuan1234/article/details/89342115