两数之和
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解题思路:
方法1:一次for循环
用一个for循环,在遍历nums的同时直接在里面查询target-nums[i]是否在nums中,如果在就输出两个数的下标,注意要判断一下两个数是否相等
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
for i in range(len(nums)):
if target-nums[i] in nums:
if i!=nums.index(target-nums[i]):
return i,nums.index(target-nums[i])
这个是能想到的比较简单的解法,但是可以看到执行用时太长,还有更加好的方法
方法2:基于Python字典查找
参考于 linfeng886的博客
先建立一个空字典dict{},遍历nums
如果nums[i]不在字典中就把target-nums[i]的值存入字典,(target-nums[i]:i),也就是把nums中每个数与target对应的差值和该数的下标存入字典中
如果nums[i]在字典中,就说明nums中前面的数找到了它的另一半(与target的差值),输出就可以了
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
dic = {} #创建一个空字典
for i in range(len(nums)):
a = target-nums[i]
if nums[i] in dic:
return dic[nums[i]],i
else:
dic[a] = i #往字典中添加键值对
ps:第一次刷题笔记,希望不是最后一次,嘻嘻