问题:
给定一个数和一个数列,找到这个数列中某两个元素之和为给定的数,返回这两个数的 index
代码
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
for i,num in enumerate(nums):
sub_num = target-num
if sub_num in nums:
j = nums.index(sub_num)
if j<i:
return [i,j]
思路:
1.自然而然地想到暴力破解的思路,易得是 O()的时间复杂度,尝试了一下,时间超过了
2.时间不够,那就空间换时间,用 hashmap 的思路,会得到时间复杂度和空间复杂度都是 O(n)的解法
Note:
- enumerate 函数:enumerate(sequence,[start]=0)指定一个迭代器和一个初始位置(下标),返回一个枚举对象(enumerate)
- index函数:list.index(object)返回list 中第一个 object所在的索引,注意是第一个,所以 line 12的<可以改成!=,但不能改成>。举例,nums=[3,3],target=6。
- python的字典是用哈希实现的,所以用 enumerate 会返回一个字典- [ ]
网友评论