给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
输入:nums = [3,2,4], target = 6
输出:[1,2]
解题思路
1 暴力解法:每个元素都被枚举多次,算法时间复杂度为o(n^2)
2用 Python 中 list 的相关函数求解
解题关键主要是想找到 num2 = target – num1,是否也在 list 中,那么就需要运用以下两个方法:
num2 in nums,返回 True 说明有戏
nums.index(num2),查找 num2 的索引
def twoSum(nums, target):
lens = len(nums)
j=-1
for i in range(lens):
if (target - nums[i]) in nums:
if (nums.count(target - nums[i]) == 1)&(target - nums[i] == nums[i]):#如果num2=num1,且nums中只出现了一次,说明找到是num1本身。
continue
else:
j = nums.index(target - nums[i],i+1) #index(x,i+1)是从num1后的序列后找num2
break
if j>0:
return [i,j]
else:
return []
哈希表:类似字典,key:value
哈希解法:只要一次遍历,遍历到i时,在哈希表中查找target-i是否存在即可
若target-i存在:返回i和target-i的下标
若target-i不在:在哈希表中添加第i个元素
时间:o(n)
空间:o(n)
解题思路:哈希表
如果我们使用暴破,会导致时间复杂度为 n^2这样的代价无疑是很大的。 所以我们很容易想到用哈希表来解决这个问题。
我们遍历到数字 a 时,用 target 减去 a,就会得到 b,若 b 存在于哈希表中,我们就可以直接返回结果了。若 b 不存在,那么我们需要将 a 存入哈希表,好让后续遍历的数字使用。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashtable=dict()
for i,num in enumerate(nums):
if target-num in hashtable:
return [hashtable[target-num],i]
hashtable[num]=i
# @lc code=end
补充知识:
Python 3 新特性:类型注解
前几天有同学问到,这个写法是什么意思:
def add(x:int, y:int) -> int:
return x + y
我们知道 Python 是一种动态语言,变量以及函数的参数是不区分类型。因此我们定义函数只需要这样写就可以了:
def add(x, y):
return x + y
这样的好处是有极大的灵活性,但坏处就是对于别人代码,无法一眼判断出参数的类型,IDE 也无法给出正确的提示。
于是 Python 3 提供了一个新的特性:
函数注解
也就是文章开头的这个例子:
def add(x:int, y:int) -> int:
return x + y
用 : 类型
的形式指定函数的参数类型,用 -> 类型
的形式指定函数的返回值类型
在 Python 3.6 中,又引入了对变量类型进行注解的方法:
a: int = 123
b: str = 'hello'
更进一步,如果你需要指明一个全部由整数组成的列表:
from typing import List
l: List[int] = [1, 2, 3]
但同样,这些仅仅是“注解”,不会对代码产生任何影响。