写在前面的话
楼主最近才接触到leetcode刷题,做的第一道题,两数之和,评论上说非常简单,但是,,,哎,,,楼主感觉好难啊,终于是东拼西凑搞出来了,纪念一下,不多批批,上过程。
题目描述:
两数之和:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
解题思路:
遍历整个数组,为加快速度,将原始数组转换成为字典。
选取数组中的一个数b,用目标值减去b得到a,判断a是否在字典中,如果不在,将b和其索引成对保存到字典中,如果在,返回a在字典中对应的value(即原数组中的索引)和当前数值的索引值([ , ])。
代码:
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
d = {} # 建立空字典
for x in range(len(nums)): # 遍历原数组
b = nums[x] # 挨个取值
a = target - b # 计算剩余值
if a in d: # 判断剩余值是否在字典中
return [d[a], x] # 返回剩余值所对应的value,即其在原始数组中的索引,以及当前数值索引x
else:
d[b] = x # 将取出数值和其索引存储到字典中 dict[key] = value
nums = [2, 7, 11, 15]
target = 9
a = Solution()
out = a.twoSum(nums,target)
print(out)
执行结果:
起码算是完成了一道题,希望自己能够坚持下去!