本文主要是介绍用python手刃Leetcode(1):两数之和【简单题】,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
前言
博客里新开一个“用python手刃Leetcode”的专题,顾名思义,主要目的是记录自己在Leetcode上刷题的过程,代码全程用python3编写,所用网址是:leetcode中国。这个网址的好处是:1)纯中文,解决了英语不好读题困难的问题;2)网页打开速度快。
`
同时可以看到:目前官网给出的解题代码是用java编写的,并没有给出官方的python解答,本专栏除了尽量使用多种解法给出每道题的python3解答代码之外,还会记录下自己在做题过程中的一些收获和思考。便于以后复习查阅。
题目
给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
暴力解法
首先不管三七二十一,利用暴力解法来尝试。
class Solution:def twoSum(self, nums, target):""":type nums: List[int]:type target: int:rtype: List[int]"""renum = []n = len(nums)for i in range(n):for j in range(i+1,n):if nums[i]+nums[j] == target:renum.append(i)renum.append(j)break return(renum)
【备注】:对于一个函数来说,程序在执行return()后会自动停止,即在函数中,return有break的功能。所以,以上代码可以做一个小优化:在break处直接写return()函数。就像这样:
……
if nums[i]+nums[j] == target:renum.append(i)renum.append(j)return(renum) #此处直接用return代替break
但是,这种暴力解法由于时间复杂度太高,一般是没有办法来通过leetcode的,会报错如下:
方法二:建立哈希表进行求解。
class Solution:def twoSum(self, nums, target):""":type nums: List[int]:type target: int:rtype: List[int]"""dic = {}for i, num in enumerate(nums):if num in dic:return [dic[num], i]else:dic[target - num] = i
这个解法的思路是:首先把原来的List建成一个字典,然后新建另一个空字典,每次在字典中存储时,dic的key:target - num,value:原List组成的字典值对应的索引。如果检测到某个nums中的值已经存在于新建的字典中,那么证明可以组成一个target值,这时返回此刻遍历到的索引与dict中记录下来的索引。
方法三:利用python切片。
class Solution:def twoSum(self, nums, target):for i in range(len(nums)):if target-nums[i] in nums[i+1:]:return [i,nums.index(target-nums[i],i+1)]
解题收获:
1、enumerate()函数的使用
enumerate()函数可以把一个List按照索引从小到大的顺序组成一个字典。
python enumerate用法总结
2、python中定义Class的模板
这篇关于用python手刃Leetcode(1):两数之和【简单题】的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!