题目
给定一个整数数组 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]
提示:
(1)2 <= nums.length <= 104
(2)-109 <= nums[i] <= 109
(3)-109 <= target <= 109
(4)只会存在一个有效答案
进阶:你可以想出一个时间复杂度小于 O(n2)
的算法吗?
解题思路
法一:暴力枚举(执行用时:2160ms,内存消耗:14.1MB)
直接枚举数组中的每一个数x,寻找数组中是否存在target - x。由于每一个位于x之前的元素都已经和x匹配过了,因此只需要在x后面的元素中寻找target - x即可。
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
n = len(nums)
for i in range(n):
for j in range(i+1,n):
if nums[i] + nums[j] == target:
return [i,j]
return []
时间复杂度:O(N²),其中N是数组中的元素数量,最坏情况下数组中的任意两个数都要被匹配一次。
空间复杂度:O(1)
法二:哈希表(执行用时:28ms,内存消耗:13.5MB)
创建一个哈希表,对于每个x,我们首先检查哈希表中是否存在target - x,然后将x插入到哈希表中,即可保证不会让x和自己匹配。
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
hashmap = {}
for i,x in enumerate(nums):
diff = target - x
if diff in hashmap:
return [hashmap[diff],i]
hashmap[x] = i
时间复杂度:O(N),其中 N 是数组中的元素数量。对于每一个元素
x
,我们可以 O(1)地寻找target - x
。空间复杂度:O(N) ,其中 N 是数组中的元素数量。主要为哈希表的开销。