【题目描述】
给定一个整数数组 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
【解答】
从 i = 0 开始,依次遍历nums[i]后面的元素,判断相加是否为target
如果找到结果,返回 [i,j] ,若 没有结果返回 []
def twoSum(nums, target):
for i in range(len(nums)-1):
for j in range(i+1,len(nums)):
if nums[i] + nums[j] == target:
return [i,j]
return []
执行用时:4848ms
这是解法一的优化版,查找num1之前或之后的元素,看是否存在num2,这里选择向前遍历
若是选择查找num1之后的元素,则可能在后面的元素存在多个num2,不方便index函数返回索引
def twoSum(nums, target):
for i in range(1,len(nums)):
temp = nums[:i]
num2 = target - nums[i]
if num2 in temp:
return [temp.index(num2),i]
执行用时:792ms
用字典记录num1和num2的值&位置,省去了再查找索引的步骤
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashmap = {} #定义一个空字典
for i,element in enumerate(nums):
hashmap[element] = i #key:value格式为 nums[i]:i
for j,element in enumerate(nums):
num2 = hashmap.get(target-element)
if num2 is not None and num2 != j: #是否有一个 key 为num2 且 num2的序号不等于当前的序号j
return [j,num2]
执行用时:60ms
【总结】
所以 dict是用空间换取时间的一种方法
原文:https://www.cnblogs.com/xxx1206/p/11923911.html