题目描述:给一个整数数组,找到两个数使得他们的和等于一个给定的数target。你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是1到n,不是以0开头。
样例:numbers=[2, 7, 11, 15], target=9, return [1, 2]
可以先建立一个哈希表,键为数组中的元素,键所对应的值为数组的下标。
这样的话,我们就可以从头遍历数组,对于每一个元素,计算出目标元素与当前元素的差值,如样例中,遍历到第一个元素2时,计算出9 - 2 = 7,然后寻找7在不在哈希表里面,在的话,返回7所在的下标(当然,因为这里从1开始计数,所以还要对下标加一);不在的话,继续向后遍历数组。
代码如下:
class Solution:
"""
@param numbers : An array of Integer
@param target : target = numbers[index1] + numbers[index2]
@return : [index1 + 1, index2 + 1] (index1 < index2)
"""
def twoSum(self, numbers, target):
hash_table = {}
for i, ele in enumerate(numbers):
hash_table[ele] = i
for i, ele in enumerate(numbers):
gap = target - ele
if gap in hash_table.keys():
return [i + 1, hash_table[gap] + 1]
# write your code here
python中enumerate(枚举)能同时返回列表的
下标
和
元素。这里的哈希表用了一个字典来构造。