Leecode(72)1716.计算力扣银行的钱-简单

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