LeetCode 1.两数之和(简单)

1.题目

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

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例

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

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.解题

方法一：暴力破解-耗时甚至超时

• 拿数组的第i个数和后面的数相加，判断是否等于target。其中i从1开始直到n-1, n是数组长度。 一旦等于target了，就返回那两个数字所在的下标
• 复杂度分析：
  时间复杂度 O(Nˆ2): 因为使用了2层循环
  空间复杂度 O(1)

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        length = len(nums)
        for i in range(length):
            for j in range(i + 1, length):
                if nums[i] + nums[j] == target:
                    return [i, j]

方法二：使用数组的in和index方法

• 利用数组的in方法和List的index方法
• 遍历数组的数nums[i]，通过in方法**target-num[i]**判断是否在数组内，如果在，则使用index方法获得下标。(注意要判断是否是使用了重复的元素)

class Solution:
    def twoSum(self, nums, target):
        length = len(nums) # 数组长度
        for i in range(length): # 遍历数组
            if target - nums[i] in nums: # 判断target-nums[i]是否在nums数组中
                j = nums.index(target - nums[i])
                if j != i: # 如果j不等于i, 说明没有重复利用元素
                    return [i, j]

方法三：用字典提高查询效率

此方法参考图解答案。

• 遍历的过程中将数组存入字典中，查询操作在字典上进行，能提高速度
• 复杂度分析：
  时间复杂度 O(N)：只遍历了一次算法
  空间复杂度 O(N)：构建了一个新的字典

class Solution:
    def twoSum(self, nums, target):
        s = {} # 构建字典
        length = len(nums) # 数组长度
        for i in range(length): # 遍历数组
            if target - nums[i] in s: # 在集合中根据key值查找
                return [s[target - nums[i]], i]
            else:
                s[nums[i]] = i # 集合s中的key值存放数组值，方便查找，value值存放下标