前言
我是一枚python小白,刚开始刷Leetcode,边练习边巩固基础知识。
曾经的我自诩为学霸,其实不过是个循规蹈矩、漫无目标的三好学生。后来渐渐明白,学习方法、学习机制、学习规律的重要性。这也是我开始撰写学习笔记的初衷——记录我的学习历程,不断完善和修正。
我只是个空有激情和梦想的傻蛋,总是渴望自己能运用逻辑推理、能摆脱人云亦云、能将世上各种多元的知识都收入囊中,可是我从不付出努力和行动。
那么,就从今天开始吧!!!
题目:两数之和
给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
(1) 给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
(2) 给定nums = [3, 3], target = 6
所以返回[0, 1]
测试方法:
nums = [2,7,11,15]
target = 9
out = Solution()
output = out.twoSum(nums, target)
print(output)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
解法是网上搜集的,毕竟我是小白,现阶段以汲取高手的智慧为主,不求创新。但我对每个解法写了自己的理解,可能会有错误吧,望不吝赐教。
解法一:嵌套循环
对于包含n个元素的列表,最多需要(n-1)+(n-2)+…+ 1 = n*(n-1)/2次循环,时间复杂度为O(n^2)
class Solution:
def twoSum(self, nums, target):
n = len(nums)
for i in range(n):
for j in range(i+1, n):
if nums[i] + nums[j] == target:
return i, j
break
else:
continue
知识复习:range()
python range() 函数可创建一个整数序列的对象,一般用在 for 循环中。
函数语法:range(start, stop[, step])
计数从 start (默认是从 0 )开始,到 stop 结束(但不包括 stop)。例如:range(5) 即 range(0, 5, 1) 可以利用 list 函数返回列表list(range(5)):[0, 1, 2, 3, 4]。
执行用时 | 内存消耗 |
---|---|
6536ms | 14.4MB |
解法二:差值查找
由于list对象的存储结构为线性表,查询复杂度为O(n)。整体复杂度为O(n^2)。
class Solution:
def twoSum(self, nums, target):
for x in range(len(nums)):
b = target-nums[x]
if b in nums:
y = nums.index(b)
if y != x:
return x, y
执行用时 | 内存消耗 |
---|---|
1204 ms | 14.2MB |
解法三:差值索引字典
dict对象的存储结构采用的是散列表(hash表),所以查询的时间复杂度为O(1),总体复杂度为O(n)。
class Solution:
def twoSum(self, nums, target):
dic = {}
for i, num in enumerate(nums):
if num in dic: #搜索key复杂度为O(1)
return [dic[num], i]
else:
dic[target - num] = i
执行用时 | 内存消耗 |
---|---|
44 ms | 14.8MB |
知识复习:enumerate()
enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,一般用在 for 循环当中。
函数语法:enumerate(sequence, [start=0])
seasons = [‘Spring’, ‘Summer’, ‘Fall’, ‘Winter’]
list(enumerate(seasons))
[(0, ‘Spring’), (1, ‘Summer’), (2, ‘Fall’), (3, ‘Winter’)]
另一版本:
class Solution:
def twoSum(self, nums, target):
n = len(nums)
d = {}
for x in range(n):
a = target - nums[x]
if nums[x] in d:
return d[nums[x]],x
else:
d[a] = x
执行用时 | 内存消耗 |
---|---|
44 ms | 15MB |