给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 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]
"""
for i in nums:
if target - i in nums:
return [nums.index(i), nums.index(target - i)]
nums = [2, 7, 11, 15]
target = 9
s = Solution()
print(s.twoSum(nums, target))
以上代码其实还有一点问题,就是 同样的元素不能被重复利用,假如数组是[4,3,5],target=8,那么返回值就是[0,0] 了
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
for i in nums:
if target - i in nums and i is not target-i:
return [nums.index(i), nums.index(target - i)]
# nums = [2, 7, 11, 15]
# target = 9
nums = [4, 3, 5, 15]
target = 8
s = Solution()
print(s.twoSum(nums, target))
如果这个题目只用一次循环如何找到这两个数?
思路: 我们可以将数据放在dict中
这个涉及到判断一个元素在不在一个容器中,容器的数据结构不同,查找方式不同
对于dict,其他语言的map,判断一个元素在不在容器中,list要遍历,而set和dict直接根据哈希算出,不需要遍历.
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
dict1 = {}
for k, i in enumerate(nums):
dict1[i] = k
if target - i in dict1 and i is not target - i:
return [dict1[target - i], dict1[i]]
# nums = [2, 7, 11, 15]
# target = 9
nums = [4, 3, 5, 15]
target = 8
s = Solution()
print(s.twoSum(nums, target))
点评:其实以上代码还是存在问题的,问题在于,对于相同的元素[3,3],target=6,测试一下,便懂.
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
for k, i in enumerate(nums):
if target - i in nums[k + 1:]:
return [k, nums[k + 1:].index(target - i) + k + 1]
那么对于字典的那种方式,就只能,索引为key,数据为value,只是这样一来,判断在或者不在,还是多了一层循环