python百题大冲关-查找两个总和为特定值的索引

DaLongggggg 等级 125 0 0

本次挑战中,你需要在 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
预览图
收藏
评论区