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;
    }
};