[LeetCode刷题记录]两数之和

最新推荐文章于 2024-01-21 17:48:40 发布

清水寺丞

最新推荐文章于 2024-01-21 17:48:40 发布

阅读量71

点赞数

分类专栏：刷题心得文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/m0_56738500/article/details/126574215

版权

刷题心得专栏收录该内容

6 篇文章 0 订阅

订阅专栏

一、题目

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

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

你可以按任意顺序返回答案。



示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路与代码

1.暴力枚举法

即暴力测试每种两个数字组合的代表

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

结果正确，但是速度并不理想,为1896ms

时间复杂度:O(n^2)

2.哈希法

在之前的题目中发现，每次查找 target-num这个数是否存在时需要O（n）的时间

如果使用哈希表，则可以将查找时间缩短到O（1）

代码如下

class Solution():
    def twoSum(self,nums, target):
        hashmap={}
        for index,num in enumerate(nums):
            hashmap[num]=index
        print(hashmap)
        for index,num in enumerate(nums):
            anotherIndex=hashmap.get(target-num,False) 
            if(anotherIndex and anotherIndex!=index):#第二个条件是防止出现一个数为target的1/2的情况
                return [index,hashmap[target-num]]

只使用了28ms