题目
给定一个整数数组 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]提示
一句话思路:用哈希表记录「每个数字出现过的下标」,遍历时查找 target - 当前数 是否已经在哈希表中。
- 为什么用哈希表:能把「查找另一个数」从 O(n) 降到 O(1),整体时间复杂度 O(n)。
- 为什么边遍历边存:保证不会使用两次相同元素(先查再存,避免把当前元素自己当成答案)。
答案
python
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
dct = {} # 哈希表:记录「值 -> 下标」,用于 O(1) 查找
for i, n in enumerate(nums): # 遍历数组,i 是下标,n 是当前值
cp = target - n # 计算需要的另一个数(target 减去当前值)
if cp in dct: # 之前遍历过这个数,说明找到配对
return [dct[cp], i] # 返回「之前那个数的下标, 当前下标」
dct[n] = i # 没找到,先把当前值和下标存入哈希表
留言板