Hercy 想要为购买第一辆车存钱。他 每天 都往力扣银行里存钱。
最开始,他在周一的时候存入 1
块钱。从周二到周日,他每天都比前一天多存入 1
块钱。在接下来每一个周一,他都会比 前一个周一 多存入 1
块钱。
给你 n
,请你返回在第 n
天结束的时候他在力扣银行总共存了多少块钱。
示例 1:
输入:n = 4 输出:10 解释:第 4 天后,总额为 1 + 2 + 3 + 4 = 10 。
示例 2:
输入:n = 10 输出:37 解释:第 10 天后,总额为 (1 + 2 + 3 + 4 + 5 + 6 + 7) + (2 + 3 + 4) = 37 。注意到第二个星期一,Hercy 存入 2 块钱。
示例 3:
输入:n = 20 输出:96 解释:第 20 天后,总额为 (1 + 2 + 3 + 4 + 5 + 6 + 7) + (2 + 3 + 4 + 5 + 6 + 7 + 8) + (3 + 4 + 5 + 6 + 7 + 8) = 96 。
class Solution:
def totalMoney(self, n: int) -> int:
# 计算完整的周数和剩余天数
full_weeks = n // 7
extra_days = n % 7
# 计算完整周数的总存款
# 每周存款总和是 28
# 每周存款量增加是 7 * (week_index)
total = full_weeks * 28 + 7 * (full_weeks * (full_weeks - 1)) // 2
# 计算剩余天数的存款
# 剩余天数的起始存款是 full_weeks + 1
start_amount = full_weeks + 1
total += sum(start_amount + i for i in range(extra_days))
return total