150天带你用python爆刷Leetcode（day1）--1. 两数之和

欢迎来到python刷题系列，今天的主角是leetcode的第一题：1，两数之和！

乍一看题目可能会觉得这只是一道非常基础的题目，确实，这道题目比较简单，但并不是如你想的简单求 a+b 这般，题目详情如下：

传送门：力扣

1. 两数之和
给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

 
示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]
示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]
 

提示：

2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案

        首先我们来仔细阅读一下题目，要求是从列表中获取指定的元素下标，并且要求这两个元素之和为target，关键在于

        “你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现”

这句话，这表明：结果答案只有一个，并且这个答案不会是元素本身和本身之和，例如有一个列表为[ 3, 3]，同时target=6，那么答案就不会是[0, 0] 或者 [1, 1]，但可以是[0, 1] 或者是 [1, 0]，这个时候第一个想到的肯定是暴力遍历了--设置快慢指针

        从头开始一步步遍历，直到nums[left] + nums[right] == target，此时跳出循环，返回答案

自然这不失为一种做法，并且题目数据量不大，确实可以完成，此时的时间复杂度：O(n) = n*(n-1) 约为 n^2，代码和提交结果如下：

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        length = len(nums)
        if length < 2:
            return None
        for left in range(length):
            for right in range(left+1,length):
                if nums[left]+ nums[right] == target:
                    return [left, right]

        显然对于这个结果肯定是不满意的，再看一眼题目，我们发现target是会提前告知的，再根据要求，只是需要求出和为target的两个数的下标，这个时候我们可以采用另外一种思路：既然target是已知的，为何不向着 num2 = target - nums1的方向转换呢？

        也就是说，使用一个循环，从头开始遍历这个列表得到nums1，则nums2 = target - nums1，但是由于题目要求必须返目标元素下标，因此这里我们使用dict来完善它，大概思路为：如果nums2不在这个dict中，就将nums1的元素和下标存入这个dict中，直到nums2出现在dict中，则返回这个nums2对应的下标和nums1此时的下标，代码和运行结果如下：

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:

        length = len(nums)
        if length < 2:
            return None
        
        dict_ = {}
        for i in range(length):
            remain = target - nums[i]
            if remain in dict_.keys():
                return [dict_[remain],i]
            else:
                dict_[nums[i]] = i
        return None

 

以上就是本篇的全部内容了，如有错误之处，欢迎斧正，评论区交给大家。