用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)]
复制代码

猜你喜欢

转载自blog.csdn.net/c710473510/article/details/89421975