Lv1. 부족한 금액 계산하기

코딩테스트 연습 > 부족한 금액 계산하기

1. 풀이 설명

price에 1부터 count까지 곱한 값에서 money를 뺀 값이 0보다 크면 구한 값을, 작으면 0을 반환한다.

2. 나의 풀이

python

def solution(price, money, count):
    total = 0
    for i in range(1, count+1):
        total += price * i
    return max(total - money, 0)

kotlin

fun solution(price: Int, money: Int, count: Int): Long {
    val result = (1..count).map { cnt ->
        price.toLong() * cnt
    }.sum() - money
    return if (result > 0) {
        result
    } else {
        0
    }
}

3. 배운점

python

1. 개선된 코드

def solution(price, money, count):
    return max(0, price * (count * (count+1))//2 -money)

kotlin

1. index나 range 를 의미있는 값으로 변경할 때 map 은 유용하다.
2. 개선된 코드: Math.max()

// asis
return if (result > 0) {
	result
} else {
	0
}

//tobe
return Math.max(result, 0)