力扣日记9.1-【哈希表篇】两数之和

最新推荐文章于 2025-12-28 22:03:12 发布

原创最新推荐文章于 2025-12-28 22:03:12 发布 · 71 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #散列表 #算法

本文介绍了如何使用哈希表解决LeetCode中的两数之和问题，通过遍历数组并在哈希映射中查找目标值减去当前元素的对应值，找到和为目标值的两个数的下标。算法的时间复杂度为O(n)，空间复杂度也为O(n)。

力扣日记：【哈希表篇】两数之和

日期：2023.9.1
参考：代码随想录、力扣

1. 两数之和

题目描述

难度：简单（www简单题都做不出来）

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

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

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

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

提示：

2 <= nums.length <= 10^4
-10^9 <= nums[i] <= 10^9
-10^9 <= target <= 10^9
只会存在一个有效答案

进阶：你可以想出一个时间复杂度小于 O(n2) 的算法吗？

题解

class Solution {
#define SOLUTION 2
public:
    vector<int> twoSum(vector<int>& nums, int target) {
#if SOLUTION == 1
        // 暴力
#elif SOLUTION == 2
        // 又没思路。。。难过
        // 看了代码随想录的思路
        // 用一个哈希map来存放遍历过的元素（存放元素及其位置）
        // 且在继续往下遍历时查询map中是否有满足的元素（即有和为target的数）
        unordered_map<int, int> record;
        for (int i = 0; i < nums.size(); i++) {
            auto entry = record.find(target - nums[i]);
            if (entry != record.end()) { // 找到了和为target的数
                return {entry->second, i};
            }
            record[nums[i]] = i;
        }
        return {0, 0};
#endif
    }
};

复杂度

时间复杂度：O(n)
空间复杂度：O(n)

思路总结

为什么用哈希法：
- 本题在遍历到一个元素 nums[i]时，要知道能否与前面的某个元素的和为 target，即要看前面是否有某个数等于 (target - nums[i]），即要看某个数是否出现过（出现在某个集合中），这就是哈希法的典型应用场景。
为什么用哈希map：
- 除了要知道某个数是否出现过，还要存储它的下标，因此用 undered_map<int, int>，存储元素及其对应下标。（不要以为哈希map只能用来计数啊kora）