携手创造,共同生长!这是我参与「掘金日新计划 8 月更文应战」的第1天,点击检查活动概况

题目描述

原题链接 :
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 <= 10^4
  • -109 <= nums[i] <= 10^9
  • -109 <= target <= 10^9
  • 只会存在一个有效答案

进阶:你能够想出一个时间复杂度小于 O(n2) 的算法吗?

思路剖析

咱们应该首先想用暴力求解的话,怎么做?

咱们会遍历数组 a,然后看除了 a 数组中有没有 target-a 的数,这样就能确保该数组有两个数和等于 target;可是时间复杂度为 O(n2)O(n^2)

接下来,想一想怎么更快一点呢?

对,咱们能够借用哈希(或许叫字典),咱们遍历元素的时分,且记录元素的下标,当咱们找 target-a 时分,只需要在字典找,就能够了,查找字典时间复杂度为 O(1)O(1)

AC 代码

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

参阅

两数之和的四种解法的六个完成(排序+减半查找/封装+排序+减半查找/暴力/哈希) – 两数之和 – 力扣(LeetCode

用刺进哈希表的方法处理,时间复杂度为On – 两数之和 – 力扣(LeetCode)