两数之和
前言
经过两个月的思考,还是放弃读博,决定去找工作,在与同学交流之后选择后端开发而不是AI算法岗,一方
面学校是个普本,另一方面自己也怕没兜底就尴尬了,开始做leetcode题还是很有必要的,两年的时间足够了。
正片
题目:
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值target的那两个整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。你可以按任意顺序返回答案。
首次答案:
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
solution=[]
for i,value in enumerate(nums):
for j ,value1 in enumerate(nums[i+1:],start=i+1):
if value +value1 == target:
solution.append(i)
solution.append(j)
return solution
代码解析:
调用双重循环,暴力解但时间复杂度为O(n^2),进而思考是否可以改进算法,我们在每次调用的时候发现有重复探索,比方说,第一次探索完之后,我们可以把差值保存起来进而用差值作为索引去寻找(哈希表).
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hash_map={}
for i in range(len(nums)):
temp=target-nums[i]
if temp in hash_map:
return[hash_map[temp],i]
else:
hash_map[nums[i]] = i
return []
注意:
没必要去先遍历一遍存储hash,因为前后相加的一致性。前面没访问到的,后面也会访问到。