leetcode网站地址:https://leetcode-cn.com/problemset/all
leetcode(1)-两数之和
题目:
给定一个整数数组nums
和一个目标值target
,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解法
首先我们当然想到的是最简单直白的解法,代码如下:
def twoSum(nums,target):
for i in range(0,len(nums)):
for j in range(i+1,len(nums)):
if nums[i]+nums[j]==target:
return [i,j]
接下来我们可以创建几个例子检查一下:
print(twoSum([1,2,3],5))
print(twoSum([2,5,3,6,8,3],8))
print(twoSum([9,15,24,18],39))
输出为:
[1, 2]
[0, 3]
[1, 2]
结果正确,但是提交上后会发生超时现象,因为这个复杂度为O(n²)
,显然复杂度太高了,这时我们可以用enumerate
,可以遍历下标和元素,复杂度为O(n)
,代码如下:
def twoSum(nums,target):
dict = {}
for index,num in enumerate(nums):
test= target-num
if test in dict:
return [dict[test], index]
dict[num] = index
return None
再次提交,发现已经成功,图片如下:
这次的讲解到此结束,下次见!