leetcode 1# 两数之和（python）

题目

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

分析

思路1

暴力解法：两层循环遍历数组，判断两两之和是否为target
code 52ms

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums)):
            for j in range(i+1, len(nums)):
                if nums[i] + nums[j] == target:
                    return [i, j]
                    break
                else:
                    continue

时间复杂度O(n^2)

思路2

单层循环，针对每个元素在list查找是否存在差值
code

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums)):
            a = target - nums[i]
            if a in nums:
                return [i, nums.index(a)]
                break
            else:
                continue

但以上代码存在一个问题，即当list中存在两个相同的数nums[i]=nums[j]，target恰好为该数的二倍，在查询List时查到其本身便会break，返回相同的两个下标。所以需要增加判断条件，避免i==j。
code 40ms

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums)):
            a = target - nums[i]
            if a in nums:
                j = nums.index(a)
                if i == j:    # 增加判断条件
                    continue
                else:
                    return [i, j]
                    break
            else:
                continue

思路3

使用字典
遍历list的同时记录当前元素与target的差值以及元素小标，向后遍历时不断与已记录的差值进行对比。
code 44ms

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        n = len(nums)
        d = {}                           # 创建一个空字典
        for i in range(n):               # 遍历list
            a = target - nums[i]         # 记录当前元素与target的差值
            if nums[i] in d:             # 判断是否已被记录字典
                return d[nums[i]], i     # 若字典中有，说明已遍历过的元素与其之和为target
                                         #           返回：[之前遍历的元素下标，当前下标]
            else:
                d[a] = i                 # 否则写入字典

Summary

1. 关于python3中函数定义的注释 https://blog.csdn.net/Rileyang/article/details/100172983
2. list.index(a)返回list中值为a的数的下标。【多个数为a呢？？】