lc-go/solutions/17/q1716/solution.go
2026-01-13 00:14:04 +09:00

16 lines
285 B
Go

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