给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
如果暴力的话,需要用两重循环,是O(n2)。两个数之间的关系可以想到用双指针,但是双指针加快速度的前提是本身是有序的,这个时候指针可以可以两边往中间靠拢,最终将O(O2)问题简化为O(n)。但是瓶颈就是排序的O(nlogn)。
下面这个方法是最快的,只有O(n)的复杂度,因为我们对于当前遍历的每个数x,我可以用数学算出我需要匹配的数是target-x,如果有个哈希表告诉我这个数存在且他的下标是什么,这样我就直接返回了,如果如果哈希表告诉我不存在,我就把这个数及其索引存到哈希表,继续遍历,直到结束。以空间换时间
class Solution:
# 时间复杂度:O(n)
# 空间复杂度:O(n)
def twoSum(self, nums, target):
dicts = {}
for i in range(len(nums)):
if target-nums[i] in dicts:
return [dicts[target-nums[i]], i]
dicts[nums[i]] = i