本次挑战中,你需要在 sum.py 文件中补充函数 two_sum 的空缺部分。
two_sum 函数接受两个参数,nums 用于指定传入的数组,val 用于指定和的值;
two_sum 函数输出含两个索引的数组,或者 TypeError、 ValueError。
你需要补充 two_sum 函数,使 two_sum 函数可以找到数组 nums 中两个总和为 val 的值的索引。要求如下:
对于传入的数组 nums,返回总和为 val 的两个值的索引;
如果数组中没有和为目标值的元素,则返回 None。
如果传入的数组 nums 或者目标值 val 为 None,需要使用 raise 语句显示 TypeError。
如果传入的数组为空数组,需要使用 raise 语句显示 ValueError。
class Solution(object):
def two_sum(self, nums, val):
if nums is None or val is None:
raise TypeError('nums or target cannot be None')
if not nums:
raise ValueError('nums cannot be empty')
cache = {}
for index, num in enumerate(nums):
cache_val = val - num
if num in cache:
return [cache[num], index]
else:
cache[cache_val] = index
return None