LeetCode 两数之和/三数之和

文章目录

[1. 两数之和](#1. 两数之和)
[1.1 题目分析](#1.1 题目分析)
[1.2 题目代码](#1.2 题目代码)
[2. 三数之和](#2. 三数之和)
- [2.1 题目解析](#2.1 题目解析)
- [2.2 题目代码](#2.2 题目代码)

1. 两数之和

1.1 题目分析

题中的关键信息：只存在一个有效答案。暴力枚举所有二元组是最直接的想法，那么如何进一步优化？

首先，这道题要我们返回最终的数组下标，而非相应的数，因此排序做法暂时略过，因为排序显然会改变数组下标。

其次，暴力枚举二元组时，确定一个数，对第二个数的暴力枚举，如果能快速定位到该数，就可优化。所以，想到引入哈希表结构。那么哈希表，是预处理初始化好，还是在遍历过程中，一边枚举，一边维护呢？因为数组中可能存在重复元素，如果预处理，那么重复元素就会互相覆盖，所以选择一边枚举，一边维护------针对这道题，选择枚举右，维护左。

1.2 题目代码

cpp 复制代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> ret;
        unordered_map<int,int> hash;
        for(int i = 0; i < nums.size(); i++) {
            if(hash.count(target - nums[i])) {
                ret.push_back(i);
                ret.push_back(hash[target - nums[i]]);
            }
            hash[nums[i]] = i;
        }
        return ret;
    }
};

2. 三数之和

LeetCode 链接

2.1 题目解析

这道三数之和题目，在二数之和基础上，有联系，又有区别。

第一种解法，暴力枚举所有三元组，这是最直接做法。

第二种解法，固定枚举一个数。剩下问题，本质就是寻找和为特定值的二元组问题。这样分析后，结合第一道题目，肯定有办法使用哈希表解决，但是题中要求答案中不可包含重复三元组，因此，如果直接用哈希表处理，并不方便。

同时，返回的三元组，不是下标，而是具体的数，因此可以选择排序做法，这样相同的数一定是聚在一起的，可以通过指针遍历进行跳过。而在排序情况下，寻找二元组，可以使用首尾双指针相向枚举之法解决。

2.2 题目代码

cpp 复制代码

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