1. 两数之和
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
方法1:暴力遍历
思路是从列表中取一个数
a
k
a_k
ak,而后遍历
a
k
a_k
ak后的每一个数
a
j
a_j
aj,如果有:
a
k
+
a
j
=
t
a
r
g
e
t
a_k+a_j = target
ak+aj=target,则返回两数的下标。
在下面的代码中要注意
i
n
d
e
x
2
index2
index2的值是从0开始的,因此要得到在原列表中的真实下标,需要变换成
i
n
d
e
x
1
+
i
n
d
e
x
2
+
1
index1+index2+1
index1+index2+1,用时5800ms。
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
for index1,num1 in enumerate(nums):
for index2, num2 in enumerate(nums[index1+1:]):
if num1 + num2 == target:
return [index1, index1+index2+1]
方法2:两遍哈希
此方法的思想是建立一个
n
u
m
num
num和
t
a
r
g
e
t
−
n
u
m
target-num
target−num一一对应的字典,而后用dict的get方法去尝试寻找结果下标,下面的代码在建表时使用了set作为value,用时52ms。
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
# Hash
dict1 = {}
for index,num in enumerate(nums):
dict1[num] = (index,target-num)
for index1,num in enumerate(nums):
index2 = dict1.get(target-num, -1)
if index2 != -1 and index2[0]!=index1:
return [index1, index2[0]]