给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
简单做法:
数组中一个元素+另一个元素等于target即可
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
n = len(nums)
for i in range(n):
for j in range(i+1,n):
if nums[i] + nums[j] == target:
return i,j
双层循环,第一次取前面的数,第二层取第一层后面的数进行比对,全部遍历后,若nums[i]+nums[j] == target ,返回下标:i,j
进阶做法:
两数之和,其含义是target-n1=n2,即可以先取出数组中的一个数与target进行运算即可
利用哈希表进行编写
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashmap = {} # 创建字典
for indx,num in enumerate(nums): # enumerate函数分割nums列表为元素和下标
hashmap[num] = indx # 将nums列表中的值和下标索引重新写入hasmap中
for i,n in enumerate(nums):
j = hashmap.get(target-n) # 用hashmap的get方法获得target-n值的下标,并赋值给J
if j is not None and i!=j: # 进行判断并返回i,j的值
return i,j
先创建一个字典用于将原列表中的值和下标重新传入新的字典hashmap中
- enumerate函数用于将nums列表拆分成indx下标和num元素值
- enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标,一般用在 for 循环当中。
将haspmap字典中元素的值和下标一一对应,方便后续提取下标
第二次for循环并得到下标和元素
用hashmap的get方法获得target-n的值的下标,并将其赋值给j,如果最终没有得到j的值,则j为空,代表nums列表中没有与target相匹配的值
如果j不为空并且i!=j代表着下标不相等:i与j不是同一个数(例如target=6,列表中有3,6=3+3,但其中的两个三的下标相等则代表同一个数,则不符合题意,所以i和j的两个下标不能相等),最后返回i和j