题目
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
方法一:暴力解法,遍历两个循环
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
# 暴力 两个循环,分别取两个数,和为target,且索引不同,则返回
# 因为需要返回索引,所有要遍历索引,不是遍历元素
# 用.index()不行,因为如果有重复的,返回的是第一个的值
# time: O(n*2)
# space: O(1) 没用额外数组等空间
for i in range(len(nums)-1): # 下标
for j in range(i+1, len(nums)): # 下标
if nums[i] + nums[j] == target:
return i,j
方法二:用字典,空间换时间
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
# 用自字典,存储一一对应,提升速度,查找快O(1) ,空间换时
# 先表示出another_num,看其在字典中吗(在列表中慢)?在就返回;不在就加到字典中
# time:O(n) 97.52%
# space:O(n) 字典存储,最大长度等于nums长度
if not nums:
return None
d = {}
for i, num in enumerate(nums):
ano_num = target - num
if ano_num in d:
# 如果另一个数字在字典中,且二者索引不同,即为所求
return i, d[ano_num] # error: d
d[num] = i