포스트

195. 부족한 금액 계산하기


1
2
3
4
5
6
7
8
9
10
11
12
13
14
import Foundation

func solution(_ price:Int, _ money:Int, _ count:Int) -> Int64{
    var answer:Int64 = -1
    
    var Price : Int = 0
    
    Price = (1...count).map{$0}.reduce(0, +) * price
    
    answer = money > Price ? 0 : Int64(Price-money)
    
    
    return answer
}

단순 범위를 이용하여 map으로 배열을만들고 reduce를 함으로써 간단하게 범위의 합을 구할 수 있어서 1부터 카운트 까지 합을 구하였다.

이유는 입출력의 예를 보고 이용금액이 (3+6+9+12) 라는것은 3(1+2+3+4) 이고 즉 price * count까지의 누적합이기 때문이었다.

그렇게 총 이용금액을 구해두고 삼항연산자를 통해 return 하였다.

이 기사는 저작권자의 CC BY 4.0 라이센스를 따릅니다.