前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode刷题DAY 8:两数之和

LeetCode刷题DAY 8:两数之和

作者头像
三猫
发布2020-05-08 16:20:41
3470
发布2020-05-08 16:20:41
举报

看似简单的一道题目,但要是想提升代码效率还是要费点脑筋。

1

题目描述

给定一个整数数组 nums 和一个目标值 target,在该数组中找出和为目标值的那两个整数的数组下标。如nums = [2, 7, 11, 15], target = 9时,返回[0,1](因为2+7=9)。

2

解题

思路一:遍历匹配

通过两层for循环对所有可能逐一进行验证,找到正确结果。此方法python程序在LeetCode中显示时间超出限制,不推荐。

思路二:哈希表

通过一层for循环结合enumerate()找到每个数字值与位置的关系,存储在哈希表中。因为要找两数之和,所以每次循环时,首先判断target-item是否在已经建立的哈希表中,如果在则返回两个值的位置,如果不在把本次得到的值与位置关系新添在哈希表中。

代码语言:javascript
复制
class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        h_map = {} # 或 h_map = dict()
        for i,item in enumerate(nums):
            if target-item in h_map : 
            # 或 if h_map.get(target-item) is not None: ,
            # 但速度略慢一些
                return [h_map[target-item],i]
            h_map[item]=i

需要注意的是,类似nums=[2,2],target=4的输入,生成哈希表时第二个2的位置会覆盖第一个2的位置,导致无法得到[0,1]的结果,因此要在循环最后将本次的值与位置关系加入哈希表中,而不是在两层for后首先将对应关系加入哈希表。并且本方法,判断target-item是否在列表中时,是在item之前的值中查找符合条件的值,而不是针对列表中全部元素进行判断,使得效率进一步提升(不用担心这样会错过正确答案,因为当两个元素都循环到时,结果依旧可以识别出来)。


本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-05-06,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 机器学习养成记 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档