[Algorithm] 양꼬치 (w/ Swift & Python)

알고보니 머쓱이는 양꼬치 가게의 사장님이었다.

 

 

양꼬치

문제 설명

머쓱이네 양꼬치 가게는 10인분을 먹으면 음료수 하나를 서비스로 줍니다. 양꼬치는 1인분에 12,000원, 음료수는 2,000원입니다. 정수 n과 k가 매개변수로 주어졌을 때, 양꼬치 n인분과 음료수 k개를 먹었다면 총얼마를 지불해야 하는지 return 하도록 solution 함수를 완성해보세요.

 

제한사항

• 0 < n < 1,000
• n / 10 ≤ k < 1,000
• 서비스로 받은 음료수는 모두 마십니다.

 

• solution.swift

import Foundation 

func solution(_ n:Int, _ k:Int) -> Int {    
    return (n*12000) + ((k - (n/10))*2000)
}

 

※  두 개의 정수 n과 k를 입력받아 (n * 12000) + ((k - (n / 10)) * 2000) 공식을 통해 계산한 결과를 반환하는 간결한 코드

 

func solution(_ n: Int, _ k: Int) -> Int {    
    let baseCost = n * 12000
    let discount = (k - (n / 10)) * 2000
    
    let totalCost = baseCost + discount
    
    return totalCost
}

 

※  코드의 가독성을 높이고 재사용성을 고려하여 함수 내부를 세분화 한 코드 

 

 

 

 

• solution.python

def solution(n, k):    
    baseCost = n * 12000
    
    if n >= 10:					
        discount = (k - n // 10) * 2000
    else:
        discount = k * 2000
    
    answer = baseCost + discount
    return answer

 

※  이렇게 변수명을 명확하고 직관적으로 지정하면 코드를 이해하기 쉽고 유지보수하기도 편리하다.