给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 target
的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
这道题目是一道经典的求两数之和问题,我们可以利用 Hash 表实现时间复杂度为 O(n) 的算法来解决这个问题。
具体流程如下:
1.我们以哈希表的形式存储 nums 数组中每个元素对应的下标位置。
2.然后遍历 nums 数组,计算当前值与 target 的差值 diff。
3.通过查找哈希表中的 diff 是否存在来判断是否找到了相应的答案。
4.如果没有找到,则将当前元素的值加入哈希表中。
代码实现如下:
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
# 初始化哈希表
hash_map = {}
# 遍历 nums 数组
for i in range(len(nums)):
# 计算差值
diff = target - nums[i]
# 如果在哈希表中找到了相应的差值,返回 [hash_map[diff], i]
if diff in hash_map:
return [hash_map[diff], i]
# 将当前元素的值加入哈希表中
hash_map[nums[i]] = i
时间复杂度为O(n),运行时间为64 ms,内存消耗为15.7 MB,可以 AC 这道题目。