给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。 你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。 你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9 输出:[0,1] 解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6 输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6 输出:[0,1]
提示:
核心思想:使用两个for循环来遍历该数组,数组下标不能一样,避免出现重复的可以加上标识符来退出程序。
代码如下:
class Solution: def twoSum(self, nums: List[int], target: int) -> List[int]: arr=[] flag=0 for i in range(len(nums)): for j in range(len(nums)): if nums[i]+nums[j]==target and flag!=1 and i !=j: arr.append(i) arr.append(j) flag=1 return arr
这个就是常规求法(暴力枚举求解),这样时间复杂度为O(n^2),要是对执行时间有要求估计就不行了。
核心思想:先将数组使用enumerate()函数变成索引序列,将数据保存到字典中,然后遍历求解即可!
class Solution: def twoSum(self, nums: List[int], target: int) -> List[int]: dict1={} #uid是下标,num是值 #用值作为键,不怕冲突 for uid,num in enumerate(nums): dict1[num]=uid for uid,num in enumerate(nums): #target-num是剩下多少,然后遍历 j=dict1.get(target-num) if uid!=j and j!=None: return [uid,j] return []
可能看上去有点难懂,用个例题解释一下
输入: nums=[2,3,4,4,7],target=5 1、先使用enumerate函数,第一个参数为下标,第二个参数为值 所以 下标 值 0 2 1 3 2 4 3 4 4 7 2、将值作为键,下标作为值,生成字典dict1 dict1[2]=0 dict1[3]=1 dict1[4]=2 dict1[4]=3 #这里键重复了,那么只记录后一个的值即dict1[4]=3 dict[7]=4 3、遍历enumerate函数 已知target=5,然后dict1[2]=0,那么可以知道还差值为3的数(target-2),那么我们使用get来获取3的键即可 1=dict1[3]=dict1[5-2] 那么此时下标就为0,1,返回即可!!!
用特例来测试
已知:nums=[3,3],target=6
dict1[3]=0
dict1[3]=1
∴可得出
dict1[3]=1
遍历enumerate函数
uid=0,num=3
j=dict1.get(6-3)=dict1.get(3)
此时j=1
因为uid不等于j且j不为空,所以返回即可!!!