两数之和
题目描述:给一个整数数组,找到两个数使得他们的和等于一个给定的数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 herepython中enumerate(枚举)能同时返回列表的下标和元素。这里的哈希表用了一个字典来构造。
声明:该文观点仅代表作者本人,入门客AI创业平台信息发布平台仅提供信息存储空间服务,如有疑问请联系rumenke@qq.com。
- 上一篇: Python--内存管理
- 下一篇:没有了