leetcode刷题第一天-第一题 两数之和
两数之和
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。你可以按任意顺序返回答案。
示例
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
输入:nums = [3,2,4], target = 6
输出:[1,2]
输入:nums = [3,3], target = 6
输出:[0,1]
拿到题目第一反应是可以通过暴力的遍历来解决,通过双重嵌套循环遍历列表,相加等于target得到数字,使用i j作为哨兵,输出哨兵的值即可 时间复杂度 n2 不继续。
思考一下 可以使用字典来进行,将数值作为键,下标作为值,将键值对存储在字典中,再从列表中遍历如果target-遍历的num 是字典中的键,则可得到答案 进行编码 提交后57个样例通过50个。
错误原因:未考虑到字典的键是唯一存在的 如果nums中有两个一样的值第一次写入的值会被更新,且在遍历的时候,如果有一个值是target的一半会出现num+num=target的错误
更改:在比较的时候加入判断语句 解决错误
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashmap={}
for ind,num in enumerate(nums):
hashmap[num] = ind
for ind,num in enumerate(nums):
if target-num in hashmap:
if ind==hashmap[target-num]:
pass
else:
res = [ind,hashmap[target-num]]
break
return res
改进:在作完后看大佬提交的代码,得到改进的思路,在原代码中,每一个数值都不可避免的和自己进行了比较,可以在生产字典的同时进行比较,既倒着比较,每个数值只会让他前面的数进行比较,并在比较后将自己加入到字典中, 减少比较次数,提高效率同时可以减少一次判断的情况下解决相等和半值问题
代码如下
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashmap={}
for ind,num in enumerate(nums):
if target-num in hashmap:
return [hashmap[target-num],ind]
hashmap[num] = ind