一、题目
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
二、题解
1.暴力枚举
class Solution:
def twoSum(self, nums, target) :
n = len(nums)
# 枚举在数组中所有的两个不同的下标的组合
# 左闭右开,到n即可
for i in range(n):
for j in range(i + 1, n):
if nums[i] + nums[j] == target:
return [i, j]
return []
l = Solution()
print((l.twoSum([], 27))) # []
print(l.twoSum([5, 6], 7)) # []
print(l.twoSum([5, 6], 11)) # [0, 1]
print(l.twoSum([2, 7, 11, 15], 9)) # [0, 1]
print(l.twoSum([5, 7, 11, 15], 26)) # [2, 3]
2.哈希表法
创建一个哈希表,对于每一个 x,我们首先查询哈希表中是否存在 target - x,然后将 x 插入到哈希表中,即可保证不会让 x 和自己匹配。
如:
注意:这里是将下标作为值value来存入哈希表
6之前没有对应的元素,直接将6和下标0存入哈希表
下一个元素是3,target-3=5,5不在哈希表中,将当前遍历到的元素3和对应下标1存入哈希表
下一个元素是8,target-8=0,0不在哈希表中,将当前遍历到的元素8和对应下标2存入哈希表
下一个元素是2,target-2=6,6在哈希表中,6和2就是要找的两个元素,对应的下标分别是哈希表中6对应的下标0和当前遍历到的元素的下标3,输出[0, 3]
class Solution:
def twoSum(self, nums, target):
# 创建一个字典,python中字典使用哈希表实现,哈希表的中的值用来存储下标
hashtable = dict()
# 遍历,i为nums数组下标,cur为nums数组中的数
for i, cur in enumerate(nums):
# 如果目标值target-当前数cur在哈希表中
if target - cur in hashtable:
# 返回[(目标值target-当前数cur)的下标, 当前数cur的下标]
return [hashtable[target - cur], i]
# 否则,将元素和对应的下标存入哈希表
# 哈希表中key为数组中的数,value为数组中的数对用的下标
hashtable[nums[i]] = i
return []
l = Solution()
print(l.twoSum([6, 3, 8, 2, 1], 8)) # [0, 3]
print((l.twoSum([], 27))) # []
print(l.twoSum([5, 6], 7)) # []
print(l.twoSum([5, 6], 11)) # [0, 1]
print(l.twoSum([2, 7, 11, 15], 9)) # [0, 1]
print(l.twoSum([5, 7, 11, 15], 26)) # [2, 3]
注:
list = [1, 2, 3, 'a', 'b']
# 带引索的遍历数组
for index, data in enumerate(list):
print(index, data)
'''
相当于:
for i in range(0, len(list)):
print(i, list[i])
'''
时间复杂度:O(N),其中 N 是数组中的元素数量。对于每一个元素 x,我们可以 O(1) 地寻找 target - x。
空间复杂度:O(N),其中 N 是数组中的元素数量。主要为哈希表的开销。