青训营X豆包MarsCode 技术训练营｜豆包MarsCode AI 刷题python 复制函数签名：定义了一个名

python

复制

def solution(a: int, b: int) -> int:
    # 计算回本需要的天数，若a能被b整除则直接返回a // b，否则返回a // b + 1
    return a // b if a % b == 0 else a // b + 1

函数签名：
- 定义了一个名为 solution 的函数，它接受两个整数参数 a 和 b。这里使用了类型提示（a: int 和 b: int）来表明期望传入的参数类型应该是整数，虽然 Python 在运行时不会强制要求这种类型，但它可以帮助提高代码的可读性以及在一些静态分析工具中发挥作用。函数的返回值类型也被指定为整数（-> int）。
函数功能：
- 函数的目的是计算回本需要的天数。具体计算逻辑是，如果 a 能被 b 整除（通过 a % b == 0 判断，其中 % 是取模运算符，用于获取除法运算的余数），那么就直接返回 a 除以 b 的整数商（a // b，这里的 // 是地板除运算符，它会返回除法运算的整数部分）。
- 如果 a 不能被 b 整除，那么就返回 a 除以 b 的整数商再加 1（a // b + 1）。

收起

python

复制

if __name__ == '__main__':
    assert solution(10, 1) == 10
    assert solution(10, 2) == 5
    assert solution(10, 3) == 4

if __name__ == '__main__': 语句：
- 这是 Python 中常见的一种结构。当一个 Python 脚本被直接运行时，__name__ 变量的值会被设置为 '__main__'；而当这个脚本被作为模块导入到其他脚本中时，__name__ 的值会是模块的名字而不是 '__main__'。所以这段代码的作用是，只有当这个脚本被直接运行时，才会执行下面缩进的代码块。
断言语句（assert） ：
- 这里使用了三个断言语句来对 solution 函数进行简单的测试。
- assert solution(10, 1) == 10：调用 solution 函数传入参数 10 和 1，并断言函数返回值应该等于 10。因为 10 能被 1 整除，按照函数逻辑 10 // 1 确实等于 10，如果函数实现有误，这里就会抛出异常。
- assert solution(10, 2) == 5：同理，传入 10 和 2，因为 10 能被 2 整除，10 // 2 等于 5，所以断言函数返回值应该是 5。
- assert solution(10, 3) == 4：传入 10 和 3，10 不能被 3 整除，按照函数逻辑 10 // 3 等于 3（取整），再加上 1 就是 4，所以断言函数返回值应该为 4。这些断言语句可以帮助在开发过程中快速发现函数实现是否符合预期的功能。

总体来说，这段代码定义了一个简单的函数用于特定的计算（回本天数计算），并通过主程序中的断言语句对函数进行了初步的测试验证。

在代码中添加异常处理

青训营X豆包MarsCode 技术训练营｜ 豆包MarsCode AI 刷题