python函数与代码复用

Question

某公司以一定的策略投资一支股票，每天都会投入一定的资金用于购买股票,其策略为:在周一的时候投入1万块钱。从周二到周日,每天都比前一天多投入1万块钱。在接下来每一个周一,都会比前一个周一多投入1万块钱。
要求:使用函数,实现对于任意输入的正整数n,能够得到在第n天结束的时候该公司在该股票上总共投入了多少钱。

Answer 1

def total_investment(n):
    monday_investment = (n // 7 + 1) * 10000
    other_investment = (n // 7) * (n // 7 + 1) // 2 * 6 * 10000
    total = monday_investment + other_investment
    return total
n = int(input("请输入天数："))
total = total_investment(n)
print("在第%d天结束的时候该公司在该股票上总共投入了%d元。" % (n, total))