[Kotlin] 부족한 금액 계산하기 (프로그래머스 알고리즘)

 

 

 

 

 

문제

https://school.programmers.co.kr/learn/courses/30/lessons/82612

 

 

 

---

 

접근

 

class Solution {
    fun solution(price: Int, money: Int, count: Int): Long {
        
        var fullPrice = 0L
        for (i in 1..count){
            fullPrice += (i * price)
            }
        //1. 총 금액을 구한다.
        
        if(fullPrice > money) return (fullPrice-money).toLong()
        else return 0L
        
         //2. 현재 자신의 금액과 비교해서 얼마가 모자라는지를 return 한다.
         //3. 모자라지 않은 경우 0을 return 한다.
        }
        
    }
}

• 필요한 총 금액을 구하고, 현재 가진 금액과 비교하는 방식으로 구상했다.
• 그러나 '/Solution.kt:18:1: error: expecting a top level declaration' 에러가 계속 떴다.
• 알고보니 별 문제는 아니고, 밑에 }를 하나 더 추가해버린 문제였다.

 

 

---

 

성공 코드

 

class Solution {
    fun solution(price: Int, money: Int, count: Int): Long {
        var fullPrice = 0L
        for (i in 1..count) {
            fullPrice += i * price
        }
        if (fullPrice > money) return fullPrice - money
        else return 0L
    }
}

 

---

 

설명

 

1. fullPrice라는 Long형 변수를 선언한다.
2. 1부터 전체 개수까지 반복하며 금액을 fullPrice에 계속 더해준다.
3. 구해진 총 금액(fullPrice)가 가지고 있는 money보다 크고 작음에 따라 return값을 분리한다.

 

---

 

다른 방법

 

class Solution {
    fun solution(price: Int, money: Int, count: Int): Long
= (1..count).map { it * price.toLong() }.sum().let { if (money > it) 0 else it - money }
}

• map 함수를 이용하여 각 count에 맞는 금액으로 이루어진 컬렉션을 만들고 sum()으로 더한다.
• let과 it을 이용해 더해진 금액에 따라 결과를 분기한다.