题目:
给定一个整数数组 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
一般的解题思路是使用两个循环遍历,但是复杂度太高,容易超时。
另外一个思路是:
遍历数组,使用一个字典储存 value -> index,对于一个新值:
1: 如果target - value 已经存在字典当中,说明已经存在了,返回这两个值的下标。
2: 否则把这个值存入字典。
这样只需要一次遍历就能完成。
代码
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
keys = {}
for i in range(0, len(nums)):
if (target - nums[i]) in keys:
v = keys[target - nums[i]]
return [v, i]
else:
keys[nums[i]] = i
网友评论