要求
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
解题思路:我们可以发现整周的话,我们的下一周比上一周多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).