15. 三数之和

Posted 2021-10-01 这是一个很随便的名字

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
示例 2：

输入：nums = []
输出：[]
示例 3：

输入：nums = [0]
输出：[]
 

提示：

0 <= nums.length <= 3000
-105 <= nums[i] <= 105

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        unordered_map<int,int> mp;
        for(int i = 0 ;i<nums.size();i++)
        {
            mp[nums[i]]++;
        }
        vector<vector<int>> ret;
        int pre_i=-9999999,pre_j=-9999999,pre_k=-9999999;
        for(int i = 0;i<nums.size();i++)
        {
            for(int j = i+1;j<nums.size();j++)
            {
                int k = -nums[i]-nums[j];
                mp[nums[i]]--,mp[nums[j]]--;
                auto it = mp.find(k);
                if(k>=nums[j]&&it!=mp.end()&&it->second>0&&((pre_i < nums[i])||(pre_i==nums[i]&&pre_j<nums[j])||(pre_i==nums[i]&&pre_j==nums[j]&&pre_k<k)))
                {
                    ret.push_back(vector<int>({nums[i],nums[j],k}));
                    pre_i = nums[i],pre_j = nums[j], pre_k = k;
                }
                mp[nums[i]]++,mp[nums[j]]++;
            }
            
        }
        return ret;
    }
};