python leetcode刷题记录（1）-两数之和

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/weixin_42183408/article/details/88832730

leetcode网站地址：https://leetcode-cn.com/problemset/all

leetcode(1)-两数之和

题目：

给定一个整数数组nums和一个目标值target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
示例:

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

解法

首先我们当然想到的是最简单直白的解法，代码如下：

def twoSum(nums,target):
    for i in range(0,len(nums)):
        for j in range(i+1,len(nums)):
            if nums[i]+nums[j]==target:
                return [i,j]

接下来我们可以创建几个例子检查一下：

print(twoSum([1,2,3],5))
print(twoSum([2,5,3,6,8,3],8))
print(twoSum([9,15,24,18],39))

输出为：

[1, 2]
[0, 3]
[1, 2]

结果正确，但是提交上后会发生超时现象，因为这个复杂度为O(n²)，显然复杂度太高了，这时我们可以用enumerate，可以遍历下标和元素，复杂度为O(n)，代码如下：

def twoSum(nums,target):
        dict = {}
        for index,num in enumerate(nums):
            test= target-num
            if test in dict:
                return [dict[test], index]
            dict[num] = index
        return None

再次提交，发现已经成功，图片如下：

这次的讲解到此结束，下次见！