k次迭代后剩余的巧克力数量

在这个题目中，我们需要实现一个程序，模拟k次迭代后剩余的巧克力数量。

输入

我们首先需要定义输入，它应该包含以下三个参数：

• n：初始的巧克力数量
• k：要迭代的次数
• m：每个人每次可以拿走的最大巧克力数量

def chocolate(n: int, k: int, m: int) -> int:
    pass

输出

我们的程序需要返回k次迭代后剩余的巧克力数量。

思路

我们可以用一个while循环，在每次迭代中计算剩余的巧克力数量。在每次循环中，我们需要将巧克力数量减去每个人所拿走的数量，即m。

def chocolate(n: int, k: int, m: int) -> int:
    i = 0
    while i < k and n > 0:
        n -= m
        i += 1
    return n

测试

我们编写一个测试函数，用来测试我们的程序是否正确。为了方便，我们可以定义一些测试用例，用它们来检测我们的函数是否正确。

def test_chocolate():
    assert chocolate(10, 3, 2) == 4
    assert chocolate(20, 5, 3) == 5
    assert chocolate(6, 2, 2) == 2

结论

我们已经成功实现了一个模拟k次迭代后剩余的巧克力数量的程序。在这个程序中，我们使用了一个while循环来完成迭代的过程，每次迭代中的计算也很简单，减去每个人所拿走的数量即可。测试结果也证明了这个程序的正确性。