LeetCode 算法题之:TwoSum

题目

给定一个无重复的整数数组和一个目标值,找出数组中两个和为目标值的元素的索引。

举例

Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

解法

最简单的做法是使用双重循环。

def twoSum(nums, target):
    for i in range(0, len(nums)):
        for j in range(i + 1, len(nums)):
            if nums[i] + nums[j] == target:
                return [i, j]

但是时间复杂度为 O(n^2),空间复杂度为 O(1),无法通过。

改用HashTable保存列表数据,以空间换取时间

def twoSum(nums, target):
    dct = {}
    for i, x in enumerate(nums):
        dct[x] = i
    for i, x in enumerate(nums):
        j = dct[target - x] if target - x in dct else -1 #关键之处
        if j != -1 and j != i:
            return [i, j]

再进一步,在单次循环中处理

def twoSum(nums, target):
    dct = {}
    for i, x in enumerate(nums):
        j = target - x
        if j in dct:
            return [dct[j], i]
        else:
            dct[x] = i

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值