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
    评论
给定一个整数数组 nums 和一个目标值 target,要求在数组中找出两个数的和等于目标值,并返回这两个数的索引。 思路1:暴力 最简单的思路是使用两层循环遍历数组的所有组合,判断两个数的和是否等于目标值。如果等于目标值,则返回这两个数的索引。 此方的时间复杂度为O(n^2),空间复杂度为O(1)。 思路2:哈希表 为了优化时间复杂度,可以使用哈希表来存储数组中的元素和对应的索引。遍历数组,对于每个元素nums[i],我们可以通过计算target - nums[i]的值,查找哈希表中是否存在这个差值。 如果存在,则说明找到了两个数的和等于目标值,返回它们的索引。如果不存在,将当前元素nums[i]和它的索引存入哈希表中。 此方的时间复杂度为O(n),空间复杂度为O(n)。 思路3:双指针 如果数组已经排序,可以使用双指针的方来求解。假设数组从小到大排序,定义左指针left指向数组的第一个元素,右指针right指向数组的最后一个元素。 如果当前两个指针指向的数的和等于目标值,则返回它们的索引。如果和小于目标值,则将左指针右移一位,使得和增大;如果和大于目标值,则将右指针左移一位,使得和减小。 继续移动指针,直到找到两个数的和等于目标值或者左指针超过了右指针。 此方的时间复杂度为O(nlogn),空间复杂度为O(1)。 以上三种方都可以解决问题,选择合适的方取决于具体的应用场景和要求。如果数组规模较小并且不需要考虑额外的空间使用,则暴力是最简单的方。如果数组较大或者需要优化时间复杂度,则哈希表或双指针方更合适。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值