Lt1716. 计算力扣银行的钱

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 。

提示：

• 1 <= n <= 1000

输入：nums = [6]
输出：[6]

思路

推出公式：如 20，week = 2，day = 6，计算方法为，（28 + 0 + 28 + 7）+ 21 + 6 * 2。第一周 28 天，第二周 28+7 天，第三周 28+14 天，第 n 周 28+(n-1)*7 天，所以是一个首项为 21，公差为 7 的等差数列，求和公式为 28 _ (week - 1) + ((week - 1) _ (week - 2) _ 7) / 2。当 day===0 时，week 加一周，然后求完和之后，求剩余天的天数，先从 set 中拿到对应的原始元数，再根据当前是第几周，判断应该加上几元，如当前是第三周，需要加上 每天都需要 (3-1) 元，总共加了 6 _ 2 元。

解答

/**
 * @param {number} n
 * @return {number}
 */
var totalMoney = function (n) {
  const set = [1, 3, 6, 10, 15, 21, 28];
  let week = Math.ceil(n / 7);
  const day = n % 7;
  if (day === 0) week++;
  return (
    28 * (week - 1) +
    ((week - 1) * (week - 2) * 7) / 2 +
    (day >= 1 ? set[day - 1] : 0) +
    day * (week - 1)
  );
};