用哈希表求解力扣第一题 两数之和

前言

记录一下刷题历程 力扣第一题两数之和 用哈希表

---

两数之和

原题目：给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案，并且你不能使用两次相同的元素。

你可以按任意顺序返回答案。

分析

遍历一边数组，计算当前元素的补数：即让目标值target减去数组中的值。如果说哈希表中存在补数，也就是目标值减去nums数组中的值，那就说明找到了两个数他们的和等于目标值。
如果数组中没有该补数，则将当前数组对应的值nums[i]和其索引i存入哈希表中，接着遍历，直到找出对应的答案。

代码如下：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        // 获取数组的大小
        int n = nums.size();
        
        // 用于存储结果的数组，初始大小为 2
        vector<int> res(2);
        
        // 用于存储数组中元素及其索引的哈希表
        unordered_map<int, int> mp;
        
        // 遍历数组中的每一个元素
        for (int i = 0; i < n; i++) {
            // 计算当前元素的补数（即target - 当前元素的值）
            int rest = target - nums[i];
            
            // 检查哈希表中是否存在这个补数
            if (mp.count(rest)) {
                // 如果存在，说明找到了两个数，它们的和等于目标值
                // 结果数组的第一个索引是当前元素的索引
                // 第二个索引是补数的索引
                res = {i, mp[rest]};
                // 退出循环
                break;
            }
            
            // 如果哈希表中没有找到补数，则将当前元素和它的索引加入哈希表
            mp[nums[i]] = i;
        }
        
        // 返回结果数组
        return res;
    }
};

解释注释

1.int n = nums.size();

计算数组 nums 的大小，并将其存储在变量 n 中。

2.vector res(2);

创建一个大小为 2 的 vector 变量 res，用于存储两个符合条件的索引。

3.unordered_map<int, int> mp;

创建一个 unordered_map（哈希表）mp，用于存储数组中的元素值及其对应的索引。

4.int rest = target - nums[i];

计算当前元素 nums[i] 的补数 rest。补数是与当前元素加起来等于目标值 target 的值。

5.if (mp.count(rest))

检查哈希表 mp 中是否存在补数 rest。count 函数用于检查 rest 是否存在于哈希表中。

6.res = {i, mp[rest]};

如果补数存在，说明找到了两个数，它们的和等于目标值。将当前元素的索引 i 和补数的索引 mp[rest] 存储到结果数组 res 中。

7.mp[nums[i]] = i;

如果哈希表中没有找到补数，将当前元素 nums[i] 和它的索引 i 加入哈希表。

总结

以上就是利用哈希表来解这道题，这个方法的时间复杂度是 O(n)，因为每个元素被遍历一次，并且哈希表的插入和查找操作都是常数时间复杂度。