小白力扣 Hot 100 刷题记录 - 三数之和

力扣 Hot 100 刷题记录 - 三数之和

题目描述

给你一个整数数组 nums，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k，同时还满足 nums[i] + nums[j] + nums[k] == 0。请你返回所有和为 0 且不重复的三元组。

示例:

输入: nums = [-1,0,1,2,-1,-4]
输出: [[-1,-1,2],[-1,0,1]]
解释: 
- nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0。
- nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0。
- nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0。
不同的三元组是 [-1,0,1] 和 [-1,-1,2]。
注意，输出的顺序和三元组的顺序并不重要。

解题思路

方法一：双指针+哈希

可以通过双指针+哈希的方法来解决这个问题，由于刚做完两数之和，三数之和可以借鉴两数之和的方法做，唯一难点就是去重。以下是具体的思路：

1. 排序数组
   首先对数组进行排序，这样可以利用有序性来避免重复解。

2. 预处理哈希表
   使用一个键为数组值，值为数组索引的哈希表。

3. 双指针查找并去重
   双重循环遍历数组，当前值和前一个相同，则continue。

4. 将结果存入
   若找到target = 0 - nums[i] - nums[j];并且哈希表的值大于j（不往前找）结果存入数组中

代码实现

c++ 实现

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ret;
        sort(nums.begin(),nums.end());
        unordered_map<int,int> myMap;

        for (int i = 0;i < nums.size();i++) {
            myMap[nums[i]] = i;
        }

        for (int i = 0;i < nums.size();i++) {
            if (i > 0 && nums[i] == nums[i-1])//去重
                continue;

            for (int j = i+1;j < nums.size();j++) {
                if (j > i+1 && nums[j] == nums[j-1])//去重
                    continue;
                    
                int target = 0 - nums[i] - nums[j];

                if (myMap.count(target) && myMap[target] >j) {//要保证不往前找
                    ret.push_back({nums[i],nums[j],target});
                }
            }
        }
        return ret;
    }
};