LeetCode1.Two Sum两数之和(哈希表法)(Python可运行,带测试用例)

一、题目

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

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

二、题解

1.暴力枚举

class Solution:
    def twoSum(self, nums, target) :
        n = len(nums)
        # 枚举在数组中所有的两个不同的下标的组合
        # 左闭右开,到n即可
        for i in range(n):
            for j in range(i + 1, n):
                if nums[i] + nums[j] == target:
                    return [i, j]

        return []

l = Solution()
print((l.twoSum([], 27))) # []
print(l.twoSum([5, 6], 7)) # []
print(l.twoSum([5, 6], 11)) # [0, 1]
print(l.twoSum([2, 7, 11, 15], 9)) # [0, 1]
print(l.twoSum([5, 7, 11, 15], 26)) # [2, 3]

2.哈希表法

创建一个哈希表,对于每一个 x,我们首先查询哈希表中是否存在 target - x,然后将 x 插入到哈希表中,即可保证不会让 x 和自己匹配。
如:
在这里插入图片描述
注意:这里是将下标作为值value来存入哈希表
6之前没有对应的元素,直接将6和下标0存入哈希表
下一个元素是3,target-3=5,5不在哈希表中,将当前遍历到的元素3和对应下标1存入哈希表
下一个元素是8,target-8=0,0不在哈希表中,将当前遍历到的元素8和对应下标2存入哈希表
下一个元素是2,target-2=6,6在哈希表中,6和2就是要找的两个元素,对应的下标分别是哈希表中6对应的下标0和当前遍历到的元素的下标3,输出[0, 3]

class Solution:
    def twoSum(self, nums, target):
        # 创建一个字典,python中字典使用哈希表实现,哈希表的中的值用来存储下标
        hashtable = dict()
        # 遍历,i为nums数组下标,cur为nums数组中的数
        for i, cur in enumerate(nums):
            # 如果目标值target-当前数cur在哈希表中
            if target - cur in hashtable:
                # 返回[(目标值target-当前数cur)的下标, 当前数cur的下标]
                return [hashtable[target - cur], i]
            # 否则,将元素和对应的下标存入哈希表
            # 哈希表中key为数组中的数,value为数组中的数对用的下标
            hashtable[nums[i]] = i
        return []

l = Solution()
print(l.twoSum([6, 3, 8, 2, 1], 8)) # [0, 3]
print((l.twoSum([], 27))) # []
print(l.twoSum([5, 6], 7)) # []
print(l.twoSum([5, 6], 11)) # [0, 1]
print(l.twoSum([2, 7, 11, 15], 9)) # [0, 1]
print(l.twoSum([5, 7, 11, 15], 26)) # [2, 3]

注:

list = [1, 2, 3, 'a', 'b']
# 带引索的遍历数组
for index, data in enumerate(list):
    print(index, data)
'''
相当于:
for i in range(0, len(list)):
    print(i, list[i])
'''

时间复杂度:O(N),其中 N 是数组中的元素数量。对于每一个元素 x,我们可以 O(1) 地寻找 target - x。

空间复杂度:O(N),其中 N 是数组中的元素数量。主要为哈希表的开销。

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值