简单 · 哈希 · 原题

题目信息

LeetCode #1 · Two Sum

标签：数组、哈希表

题目要求

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

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

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

示例 1：

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

示例 2：

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

示例 3：

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

提示：

  • 2 <= nums.length <= 10^4

  • -10^9 <= nums[i] <= 10^9

  • -10^9 <= target <= 10^9

  • 只会存在一个有效答案

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

代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int,int> mp;
        for(int i=0;i<nums.size();i++){
            if(mp[target-nums[i]]){
                return {mp[target-nums[i]]-1,i};
            }
            mp[nums[i]]=i+1;
        }
        return {0,0};
    }
};
示例拆解

站点辅助演示：用 JavaScript 展示上方代码的状态变化，不是在浏览器里执行 C++。

先查再存，避免重复使用当前下标。表内保存下标 + 1，让下标 0 与“未找到”的 0 分开；operator[] 查询缺失键时也会插入 0。

[3, 2, 4]，target = 6：查询 3 后插入 3 → 0，再存 3 → 1；查询 4 后插入 4 → 0，再存 2 → 2；查询 2 得到 2，返回 [1, 2]。

[3, 3]，target = 6：第一次查询插入 3 → 0，再存 3 → 1；第二次查询取到 1，返回 [0, 1]。

平均时间 O(n)，空间 O(n)；哈希查找最坏情况不保证常数时间。原题保证唯一解。

返回目录