题目
【【Leetcode/Python】001-Two|【Leetcode/Python】001-Two Sum】给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
a = []
for i in range(len(nums)-1):
for j in range(i+1, len(nums)):
if target == (nums[i] + nums[j]):
a = [i,j]
break
return a
存在问题
- 直接循环存在重复计算
- 下标可以使用dict的key和value,用enumerate迭代。
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
dict = {}
for i, num in enumerate(nums):
if target - num in dict:
return [d[target - num], i]
dict[num] = i