给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
''' 方法一:暴力法 时间复杂度O(n*2),空间复杂度O(1)
new_list = []
for i in range(len(nums)):
for j in range(i+1, len(nums)):
if nums[i] + nums[j] == target:
new_list.append(i)
new_list.append(j)
return new_list
'''
# 方法二:字典代替哈希法 时间复杂度O(n),空间复杂度O(n)
hashMap = {} # 先定义一个空字典
for i in range(len(nums)): # 这个for循环用于初始化字典
hashMap[nums[i]] = i # 注意这里是原来列表中的值做字典的key
for i, value in enumerate(nums):
j = hashMap.get(target - value) # 得到可能的第二个数的标
if j is not None and j != i : # 判断是否符合条件
return [i, j]