老卫带你学---leetcode刷题(1.两数之和)

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

给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例:

给定 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]
        """
 
        return_list=[]
        for i in range(len(nums)):
            for j in range(len(nums)):
                if(i!=j and nums[i]+nums[j]==target):
                    return_list.append(i)
                    return_list.append(j)
                    return(return_list)
                

可是在运行的时候就因为时间过长而报错，想想也是，两个for循环，其时间复杂度为O(n^2),不死才怪呢。

所以最好的方法是，通过加入字典映射，将数据存入字典之中，然后遍历数组，并且用目标数减去每个单元数，然后判断是否存在于字典中。

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        arr = {}
        length = len(nums)
        for i in range(length):
            if (target - nums[i]) in arr:
                return [arr[target - nums[i]], i]
            arr[nums[i]] = i