leetcode 1716. 计算力扣银行的钱 (java & python3)

java：

只会做数学简单题 重拳出击  每日一题 芜湖 冲击一月小牌牌

class Solution {
    public int totalMoney(int n) {
        int f = 1;
        int ans = 0;
        int s = n / 7;
        int r = n % 7;
        while (s-- > 0){
            ans += (f + (f+6)) * 7 / 2;
            f++;
        } 
        while(r-- > 0){
            ans += f++;
        }
        return ans;
    }
}

python3:

class Solution:
    def totalMoney(self, n: int) -> int:
        s = n // 7
        r = n % 7
        f, ans = 1, 0
        while s > 0:
            ans += (f + (f + 6)) * 7 // 2
            f += 1
            s -= 1
        while r > 0:
            ans += f
            f += 1
            r -= 1
        return ans