leetcode刷题之two sums

1 问题叙述
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
**
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
2 代码实现
自然而然的想法,首先用for循环对nums数组进行遍历,从第一个元素开始,然后再查找数组中有没有target-nums[i]这个元素;有的话,则返回当前元素下标与另一个元素的下标,作为结果输出。代码如下:

class Solution(object):
    def twoSum(self, nums, target):
        for i in range(len(nums)):
                if (target - nums[i]) in nums:
                    return [i,nums.index(target - nums[i])]

然而,实际编程过程中出现了这么一个问题,Python提供的list.index(value,from,to) “左闭右开区间”,在不指定from,to的时候,该方法默认返回数组中第一次出现该value的index值,在case([3,4,2],6)中,应该输出[1,2],而按照前面的思路,直接输出了[0,0],这显然是不正确的

没办法,只能继续思考,如何解决这个两难的问题,突然一个想法产生,我何不对nums数组进行切片呢?即我在判断数组中是否包含 target - nums[i] 元素时,不应该对整个数组进行查找,而是应该对num[i+1:]进行查找,代码如下

class Solution(object):
    def twoSum(self, nums, target):
        for i in range(len(nums)):
            num = target - nums[i]
            if num in nums[i+1:]:
                return [i, nums.index(num,i+1)]

3 最优算法
代码思路:这个算法的核心思想是构造一个字典,字典用来存储当前元素完成target需要的元素值作为key,当前元素的index作为value。然后遍历过程中进行判断,当前元素是不是属于我想要的元素,是的话,直接输出 [字典中该元素的下标,当前遍历元素下标

class Solution(object):
    def twoSum(self, nums, target):
        nums_dict = {}
        for i, num in enumerate(nums):
            if num in nums_dict:
                return [nums_dict[num], i]
            else:
                nums_dict[target - num] = i


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值