问题描述:
给定一个整数数组 nums 和一个目标值 target,
请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
Python代码
nums = [2, 7, 11, 15]
target = 9
"""
第一种思路(最耗时间):
拿数组里的第一个数字和后面的数字分别相加,
看是否等于target;如果不等于target,那么就继续拿数组里的第二个数字和后面的数字相加;
不停的去一个个试...直到等于target,返回这2个数字所在的下标
"""
class Solution1:
def twoSum(self, nums, target):
n = len(nums) # 获取nums的长度,从此题中是4
for x in range(n): # 外层循环取出下标0,对应着数组里的第一个数字
for y in range(x + 1, n): # 内层循环取出下标0,对应着数组里的第二个数字
if nums[x] + nums[y] == target: # 如果第一个数字+第二个数字==target
return x, y # 上面的判断是对的话,那么就返回下标
break # 并停止循环
else: # 如果上面的条件不满足的话,内层for循环就会继续取出下标进行判断
continue
"""
第二种思路(用一个for循环):
直接用target减去取出的数字,看结果有没有在数组里
"""
class Solution2:
def twoSum(self, nums, target):
n = len(nums)
for x in range(n):
a = target - nums[x]
if a in nums: # 判断a有没有在nums数组里
y = nums.index(a) # 有的话,就用index获取该数字的下标(索引)
if x == y:
continue # 同样的数字不能重复用,所以这里如果是一样的数字,那么就不满足条件,跳过
else: # 否则返回结果
return x, y
break
else:
continue # 上面的条件都不满足就跳过,进行下一个循环
"""
第三种思路,用字典提高速度
把原先的数组转化成字典,通过字典去查询速度就会快很多。
"""
class Solution3:
def twoSum(self, nums, target):
d = {}
n = len(nums)
for x in range(n):
if target - nums[x] in d: # 看另外一个数字有没有在字典里
return d[target - nums[x]], x
else:
d[nums[x]] = x # 把数组里的数字作为key,下标作为value存到d字典中
a = Solution1() # 实例化类
print(a.twoSum(nums, target))