题目:
给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解决思路:
1.遍历两次列表,然后判断遍历得到的数之和是否等于目标值,是:返回,否:继续遍历
2.遍历一次列表,利用减法,判断另一个数是否在列表中,是:返回,否:继续。
二者对比,第一种思路遍历两次,时间复杂度比第二种高,故采用第二种解法。
l = []
def two_sum(nums, target):
n = len(nums)
for i in range(n):
a = target - nums[i]
if a in nums[i + 1::]:
b = nums.index(a)
l.append(i)
l.append(b)
return l
else:
continue
nums = [2, 7, 11, 19]
ret = two_sum(nums, 9)
print(ret)