(算法05/Leetcode1)两数之和

题目

两数之和:给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
提示:你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

在这里插入图片描述

思路

1.首先想到的肯定是两个for循环遍历(超时淘汰)。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums)):
            for j in range(i+1,len(nums)):
                if nums[i]+nums[j]==target:
                    return [i,j]

在这里插入图片描述
2.利用差值。因为a+b=target和different=target-a,所以different=b,且b可以在数组中找到。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums)):
            different=target-nums[i]
            if(different in nums[:i]):
                return [nums.index(different),i]

在这里插入图片描述
3.简化目前所用到的内置函数,index和切片,利用map,键对应nums[i],值对应i;每次遍历得出different再查找map的键里有没有和different所对应的键,不论有没有都将当前遍历的nums[i]存储到map里。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        nums_hash={}
        for i in range(len(nums)):
            different=target-nums[i]
            if different in nums_hash:
                return [nums_hash[different],i]
            nums_hash[nums[i]]=i

在这里插入图片描述

发布了19 篇原创文章 · 获赞 4 · 访问量 520

猜你喜欢

转载自blog.csdn.net/zhuangww05/article/details/104804639
今日推荐