题目:

1
2
3
4
5
6
7
8
9
10
11
12
给定一个整数数组 nums  和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:
给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum

题解:

1、 python

1
2
3
4
5
6
7
8
9
10
11
12
class Solution:
def twoSum(self, nums, target):
hashmap = {}
for i, val in enumerate(nums):
complement = target - val
if complement in hashmap:
return [hashmap[complement], i]
hashmap[val] = i
return []

def __init__(self, nums, target):
print(self.twoSum(nums, target))

2、 JavaScript

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
/**
* @param {number[]} nums
* @param {number} target
* @return {number[]}
*/
var twoSum = function(nums, target) {
const hashMap = {};

for (let index = 0; index < nums.length; index++) {
const element = nums[index];
const complement = target - element;
if (hashMap[complement] + 1) {
return [hashMap[complement], index];
}
hashMap[element] = index;
}
return [];
};
const res = twoSum([2, 7, 11, 15], 9);

遍历列表,目标值与当前元素相减得到符合条件的元素,查找字典,如果存在符合条件的键,返回值,否则把该元素作为键,索引作为值存到字典里,重复以上步骤,直到找到对应的两个元素,返回索引列表,若未找到,返回空数组。对比一下两种语言的执行结果,不难发现,同样的算法,python 在内存消耗和执行时间上都优于 JS。