leetcode_1716 计算力扣银行的钱

104 阅读1分钟

要求

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

提示:

  • 1 <= n <= 1000

核心代码

class Solution:
    def totalMoney(self, n: int) -> int:
        week,surplus = divmod(n,7)

        # 计算整周的结果
        n1 = 21 * week + 7 * (1 + week) * week // 2

        # 计算最后一周的结果
        n2 = (week + 1 + week + surplus) * surplus //2

        return n1 + n2

image.png

解题思路:我们可以发现整周的话,我们的下一周比上一周多7元,所以计算公式21 * week + 7 + 14 + 21 + …… + 7 * week = 21 * week + 7 * (1 + 2+ 3 + …… + week) = 21 * week + 7 * (1 + week) * week // 2,最后一周的计算周一的钱数是 week + 1,周二是week + 2 ,以此类推 week + 1 + week + 2 + …… + week + surplus = (week + 1 + week + surplus) * surplus // 2 (首项加尾项乘以项数除以2).