01.两数之和 - python

刚开始在LeetCode上刷算法题，为了让加深理解和以后用到时能够快速拾起，打算每刷一道就记录下来。

---

题目

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

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

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9

所以返回 [0, 1]

解法一：暴力法

      使用一个嵌套循环把nums列表遍历两次。优点是简单，缺点是耗时太长，时间复杂度为O(n^2)

示例

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        # 方式一
        for i in range(len(nums)-1):
            for n in range(i+1, len(nums)):
                if nums[i] + nums[n] == target:
                    return i, n

提交结果

解法二:

      用一个for循环，直接在里面查询target-nums[x]是否存在于nums列表中，考虑到in方法也是在内部遍历了一遍列表，所以速度上并没有质的提升。

示例

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        # 方式二
        # 遍历nums列表
        for i in range(len(nums)):
            # 当前值所需要的差值
            a = target - nums[i]
            # 如果列表里存在差值
            if a in nums:
                n = nums.index(a)
                # 且差值不是自己本身
                if i != n:
                    return i, nums.index(a)

执行结果

解法三：

      这是在网上学习到的方法，创建一个字典d，遍历nums的同时将 target - nums[i]:i 也就是当前值所需要的差值和当前值索引存入字典d，如果nums[i] in d 说明当前值是之前遍历过的某个值需要的差值，则返回 (d[nums[i]], i)

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        # 方式三
        # {num[i]所需差值: i, num[i]所需差值: i}
        d = {}
        # 遍历nums列表
        for i in range(len(nums)):
            # 当前值所需要的差值
            a = target - nums[i]
            # 如果当前值是前面的值所需要的差值   说明可以和前面的值相加得到target
            if nums[i] in d:
                return d[nums[i]], i
            d[a] = i

执行结果