[Swift 알고리즘] - 콜라 문제 (Programmers)

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

 

 

 

테스트 케이스는 통과하는데 제출시 우르르 실패로 떠서 뭐지(?) 싶었다.

그런데 역시 내가 문제였고.. 문제를 잘 읽어봤어야 했다..

 

처음 제출했던 답은 아래와 같다

func solution(_ a:Int, _ b:Int, _ n:Int) -> Int {
    var currentCola: Int = n
    var result: Int = 0
    
    while currentCola >= a {
        currentCola -= a
        result += 1   // 빈병을 반납하면 콜라 한병을 받는거로 구했음
        currentCola += b
    }
    return result
}

 

a갯수만큼 빈병을 가져다주면 b갯수만큼 돌려 받는다는걸 

예시에만 꽂혀서 b갯수가 아닌 한병씩 돌려받는걸로 구현을 했다가 제출하니 우르르 실패가 떴었던것.

(컴퓨터는 거짓말을 안한다.🫠)

 

 

내 풀이

func solution(_ a:Int, _ b:Int, _ n:Int) -> Int {
    var currentCola: Int = n
    var result: Int = 0
    
    while currentCola >= a {
        currentCola -= a
        result += b  // <== 수정
        currentCola += b
    }
    return result
}

 

 

b 갯수만큼 돌려받는걸로 수정하니 해결~!