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 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/calculate-money-in-leetcode-bank
2. 思路分析:
分析题目可以知道我们需要计算出n天中以7为周期的个数,例如n = 20那么以7为周期的个数为2,以7为周期的钱是很好计算的,将每一周的钱累加,下一周的钱为上一周钱加7即可,所以总的钱为:(1 + 7) * 7 // 2 * period + period * (period - 1) // 2 * 7,period为以7为周期的个数,当n = 20时period = 2,我们一开始都是以1 + 2 + ... 7计算的,而下一周为上一周的钱加7所以我们需要累加上在这么多周中多出来的7,为period * (period - 1) // 2 * 7,最后需要加上以7为周期剩余的天数,为等差数列求和:(n // 7 + 1 + n // 7 + 1 + n % 7 - 1) * n % 7 // 2,首项为n // 7 + 1
3. 代码如下:
class Solution:
def totalMoney(self, n: int) -> int:
# 模拟即可: 计算出有多少个7即可
if n <= 7:
return (1 + n) * n // 2
period, last = n // 7, n % 7
return (1 + 7) * 7 // 2 * period + period * (period - 1) // 2 * 7 + ((n // 7 + 1) * 2 + last - 1) * last // 2