某公司以一定的策略投资—支股票,每天都会投入一定的资金用于购买股票,其策略为:在第一周:周一的时候投入1万块钱。从周二到本周日,每天都比前一天多投入1万块钱。从第二周算起,在接下来每一个周一,都会比从第一个周一开始算起的每一个周一多投入1万块钱。 要求: 1、利用python编写,使用函数,实现对于任意输入的正整数n,能够得到在第n天结束的时候该公司在该股票上总共投入了多少钱。
时间: 2024-02-21 16:58:16 浏览: 64
下面是实现该功能的 Python 代码:
```python
def calculate_investment(n):
total_investment = 0
week_count = 0
day_count = 0
for i in range(1, n + 1):
total_investment += (week_count * 10000 + day_count * 10000 + 10000)
day_count += 1
if day_count > 5:
week_count += 1
day_count = 0
return total_investment
n = int(input("请输入正整数n: "))
print("该公司在该股票上总共投入了 %d 元" % calculate_investment(n))
```
上述代码中,我们定义了一个名为 `calculate_investment` 的函数,接受一个正整数 `n` 作为参数,返回在第 `n` 天结束的时候该公司在该股票上总共投入的金额。
在函数内部,我们使用三个变量来记录当前周数、当前天数以及总投入金额。接下来,我们使用一个 for 循环来遍历从 1 到 n 的所有天数,对于每一天,根据题目所述的投资策略计算出当天的投入金额,累加到总投入金额中。最后,返回总投入金额即可。
在程序的最后,我们通过 input 函数获取用户输入的正整数 n,然后调用 calculate_investment 函数来计算总投入金额,并输出结果。
阅读全文