17 lines
392 B
Go
17 lines
392 B
Go
// Package q1716 implements a solution for https://leetcode.com/problems/calculate-money-in-leetcode-bank/
|
|
package q1716
|
|
|
|
func totalMoney(n int) int {
|
|
if n <= 7 {
|
|
return (1 + n) * n / 2
|
|
}
|
|
base := 28
|
|
|
|
rem := n % 7
|
|
fullWeeks := n / 7
|
|
moneyFW := (base-7)*fullWeeks + 7*(1+fullWeeks)*fullWeeks/2
|
|
moneyRem := (1+rem)*rem/2 + rem*fullWeeks
|
|
return moneyFW + moneyRem
|
|
}
|
|
|
|
var _ = totalMoney
|