python如何高效解答啊

Question

孙悟空拥有无限分身，假设他每秒钟都会产生一个分身

但第一次分身时，具有2秒CD (第1秒产生，第4秒才能使用分身术) [之后每秒都可分身] ，即

第1秒仅有本体,共1个

第2秒则是本体+分shen1共2个

第3秒则是本体+分身1+分shen2共3个

第4秒则是本体+fen身1+分shen2+fen身3共4个

第5秒则是本体+分shen1+分身2+分shen3+分身4+分身1_ 1共6个

请编程实现:求在第n秒的时候，共有多少只猴?

输入时间整数t

输出美猴王的总数sum

注释：分身重复用拼音代替了

Answer 1

基于Monster 组和GPT的调写：

def count_monkeys(n):
    if n == 1:
        return 1
    elif n == 2:
        return 2
    else:
        cd = 2  # 分身术的CD
        monkeys = 2  # 第2秒时有2只猴（本体和分身1）
        new_monkeys = 1  # 上一秒产生的新猴子数
        for i in range(3, n+1):
            if cd > 0:
                cd -= 1
            else:
                new_monkeys += 1
                cd = new_monkeys
            monkeys += new_monkeys
        return monkeys

t = int(input("请输入时间整数t: "))
sum = count_monkeys(t)
print("在第%d秒的时候，共有%d只猴" % (t, sum))
print("美猴王的总数为：", sum)

运行程序后，用户输入时间整数t，程序会调用count_monkeys函数计算在第n秒的时候有多少只猴，并输出结果。该函数通过模拟孙悟空使用分身术的过程来计算猴子的数量，使用一个变量cd来表示当前分身术的CD，使用一个变量new_monkeys来记录上一秒产生的新猴子数，以及一个变量monkeys来记录当前的猴子数量。程序使用for循环遍历从第3秒到第n秒的过程，在每秒更新cd和new_monkeys，并计算出当前的猴子数量。最后，程序输出在第n秒的时候有多少只猴。