15. 3Sum 题解

15. 3Sum  题解

题目描述：

Given an array S of n integers, are there elements a, b, c in S 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.

For example, given array S = [-1, 0, 1, 2, -1, -4],A solution set is:[  [-1, 0, 1],  [-1, -1, 2]]

题目链接：15. 3Sum

算法描述：

        由题意知，给定一个数组，找到数组中所有的唯一三元组，使它们的和为零。并且要求返回的数组满足递增。

        首先，构造一个结果容器 ans。如果原数组为空，返回 ans。

        我们对原数组进行从小到大排序，之后我们对其遍历，标定三元组的首元素 i（首元素一定不大于零并且位置小于数组长度减二），之后用 l 和 r 分别从首元素的后一位（i+1）和数组的最后一位（ nums.size() - 1 ）向中间逼近。当发现三者之和为零时，将它们放入结果容器 ans。

         如果三者之和大于零，r--；如果三者之和小于零，l++。

         注意在用标记i，l 和 r 遍历及查找时需要跳过重复元素。

         算法复杂度为O(n²)

         

代码：

class Solution {public:    vector<vector<int>> threeSum(vector<int>& nums) {        vector<vector<int>> ans;        if(nums.size() == 0){            return ans;        }        sort(nums.begin(),nums.end());                for(int i=0; i < nums.size()-2 && nums[i] <= 0; i++){            if(i>0 && nums[i] == nums[i-1]){                continue;            }            int l = i+1;            int r = nums.size()-1;                        while(l<r){                if(nums[i] + nums[l] + nums[r] == 0){                    ans.push_back({nums[i], nums[l], nums[r]});                    while(++ l < r && nums[l] == nums[l - 1]);                    while(l < -- r && nums[r] == nums[r + 1]);                }                else if (nums[i] + nums[l] + nums[r] > 0){                    --r;                }                else{                    ++l;                }            }        }                return ans;    }};