(1)LeetCode 1. 两数之和

LeetCode 1. 两数之和

题目描述

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

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

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

示例:

复制代码
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1]。

解题思路

最直观的方法是使用两层循环暴力枚举,时间复杂度 O(n²)。为了优化,我们可以利用哈希表(unordered_map)将查找时间从 O(n) 降低到 O(1)。

核心思想:一边遍历数组,一边将已经遍历过的元素存入哈希表 。对于当前元素 nums[i],计算其需要的配对数 r = target - nums[i],然后检查哈希表中是否已经存在 r

  • 如果存在,说明之前出现过某个数恰好与当前数之和为 target,直接返回两个下标(之前那个数的下标和当前下标 i)。
  • 如果不存在,则将当前元素 nums[i] 及其下标 i 存入哈希表,继续遍历。

由于题目保证有唯一解,因此遍历过程中一定会找到答案。

代码实现(C++)

cpp 复制代码
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        // 创建哈希表,key 为数组元素值,value 为该元素在数组中的下标
        unordered_map<int, int> heap;
        
        // 遍历数组
        for (int i = 0; i < nums.size(); i++) {
            int r = target - nums[i];           // 计算当前元素需要的另一个数
            
            // 如果需要的数已经在哈希表中,说明找到了答案
            if (heap.count(r)) {
                return {heap[r], i};            // 返回之前的下标和当前下标
            }
            
            // 否则将当前元素存入哈希表,供后续元素查找
            heap[nums[i]] = i;
        }
        
        // 根据题意,不会执行到这里,但函数需要返回值,故返回空数组
        return {};
    }
};

复杂度分析

  • 时间复杂度:O(n),其中 n 是数组的长度。我们只遍历了一次数组,每次哈希表的查找和插入操作都是 O(1)。
  • 空间复杂度:O(n),哈希表最多需要存储 n 个元素。

注意事项

  1. 哈希表为什么用 unordered_map
    unordered_map 基于哈希表实现,平均情况下插入和查找的时间复杂度为 O(1),而 map 基于红黑树实现,是 O(log n),因此本题选用 unordered_map 更高效。

  2. 为什么先检查再插入?

    如果将当前元素先插入哈希表再检查,那么对于 nums = [3,3]target = 6 的情况,遍历第一个3时插入哈希表,第二个3查找时会在哈希表中找到自己(即同一个元素用了两次),导致错误。因此必须先检查再插入,确保不会重复使用同一元素。

  3. 返回值 {heap[r], i} 的顺序任意,题目允许任意顺序返回两个下标。

  4. 题目保证有解 ,因此不需要考虑找不到答案的情况,但为了函数完整性,代码最后保留了 return {}

总结

本题是哈希表的经典应用,通过空间换时间,将两数之和问题优化到线性时间复杂度。掌握这种"边遍历边记录"的思路,可以解决很多类似的问题,例如"三数之和"、"四数之和"的部分优化。

相关推荐
6Hzlia9 分钟前
【Hot 100 刷题计划】 LeetCode 24. 两两交换链表中的节点 | C++ 精准指针舞步
c++·leetcode·链表
沉默-_-2 小时前
备战蓝桥杯-哈希
c++·学习·算法·蓝桥杯·哈希算法
踩坑记录3 小时前
leetcode 92. 反转链表 II 区间反转(不是整条链表反转)
leetcode·链表
AI人工智能+电脑小能手4 小时前
【大白话说Java面试题】【Java基础篇】第19题:HashMap的key如何减少发生哈希冲突
java·开发语言·后端·面试·哈希算法·hash-index·hash
6Hzlia4 小时前
【Hot 100 刷题计划】 LeetCode 148. 排序链表 | C++ 归并排序自顶向下
c++·leetcode·链表
im_AMBER4 小时前
Leetcode 162 除了自身以外数组的乘积 | 接雨水
开发语言·javascript·数据结构·算法·leetcode
啊哦呃咦唔鱼4 小时前
leetcodehot100-347. 前 K 个高频元素
数据结构·算法·leetcode
玛丽莲茼蒿4 小时前
Leetcode hot100 多数元素【简单】
算法·leetcode·职场和发展
AbandonForce4 小时前
Map类:pair键值对|map的基本操作|operator[]
开发语言·c++·算法·leetcode
田梓燊4 小时前
力扣:146.LRU 缓存
算法·leetcode·缓存