题目描述
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
- Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
代码实现
1.方法一:
- 思路:这道题很容易想到用暴力解决。最简单的循环嵌套,时间复杂度是O(n*n)。
- 代码:
def twoSum(self, nums, target):
#输出结果创建list存放
result=[]
for i in range(len(nums)):# 循环遍历nums
for j in range(i+1,len(nums)):
if nums[i]+nums[j]==target:
result.append(i)
result.append(j)
return result
2.方法二:
- 思路
python字典的key-value原理属于hashtable的范畴,Python中最常用的数据类型之一,它是一个键值对的集合,字典通过键来索引,关联到相对的值,理论上它的查询复杂度是 O(1) 。
在这里我们通过建立字典,字典存储当前元素完成target需要的元素值作为key,当前元素的index作为value。然后遍历过程中进行判断,当前元素是不是属于我想要的元素,是的话,直接输出 [字典中该元素的下标,当前遍历元素下标] 。
其次,分析一下该方法的时间复杂度,第一层for循环是O(n),所以在 x in dict的查找过程中,时间复杂度为O(1),而这里没用到index查询,而是直接获取value值,字典的get item时间复杂度同样是O(1),所以整个算法时间复杂度O(n),但是这里利用空间复杂度为O(n),用空间换取了时间。
def twoSum(self, nums, target):
nums_dict = {} #创建字典dict key:当前元素完成target需要的元素值 value:当前元素的index
for i, num in enumerate(nums):# enumerate 枚举,遍历返回nums的index以及数值
if num in nums_dict: # 判断是否满足target
return [nums_dict[num], i] #返回[字典中该元素的下标,当前遍历元素下标]
else:
nums_dict[target - num] = i
总结
- 1.暴力算法也叫蛮力算法,之所以称为暴力,是因为该算法是枚举当前所有出现的情况,
从而得到需要的情况。该算法可以求的一些情况较少的问题的解,若问题规模太大,该算法便不适用。 - 2.enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标,一般用在 for 循环当中。
enumerate(sequence, [start=0])
sequence – 一个序列、迭代器或其他支持迭代对象。
start – 下标起始位置。
返回 enumerate(枚举) 对象
实例一:
>>> list(enumerate(seasons))
[(0, 'Spring'), (1, 'Summer'), (2, 'Fall'), (3, 'Winter')]
>>> list(enumerate(seasons, start=1)) # 下标从 1 开始
[(1, 'Spring'), (2, 'Summer'), (3, 'Fall'), (4, 'Winter')]
实例二:
for 循环使用 enumerate:
>>> for i, element in enumerate(seq):
... print i, element
...
0 one
1 two
2 three
感想
今天是第一天开始刷题,作为非科班、却要努力成为一名程序猿的我来说有点吃力,但是通过自己思考以及借鉴别人的想法收获了不少新的知识,我会在这条路上坚持下去的的!