用python手刃Leetcode(1)：两数之和【简单题】

分手后的思念是犯贱 2022-01-20 03:07 160阅读 0赞

> ##### 前言 #####
> 
> 博客里新开一个“用python手刃Leetcode”的专题，顾名思义，主要目的是记录自己在Leetcode上刷题的过程，代码全程用python3编写，所用网址是：[leetcode中国][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用法总结][python enumerate]

###### 2、python中定义Class的模板 ######

[leetcode]: https://link.juejin.im?target=https%3A%2F%2Fleetcode-cn.com%2F
[python enumerate]: https://link.juejin.im?target=https%3A%2F%2Fblog.csdn.net%2Fchurximi%2Farticle%2Fdetails%2F51648388